mirror of
https://github.com/Xevion/exercism.git
synced 2025-12-06 09:14:59 -06:00
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
import unittest
|
|
|
|
from collatz_conjecture import steps
|
|
|
|
|
|
# Tests adapted from `problem-specifications//canonical-data.json` @ v1.2.1
|
|
|
|
|
|
class CollatzConjectureTest(unittest.TestCase):
|
|
def test_zero_steps_for_one(self):
|
|
self.assertEqual(steps(1), 0)
|
|
|
|
def test_divide_if_even(self):
|
|
self.assertEqual(steps(16), 4)
|
|
|
|
def test_even_and_odd_steps(self):
|
|
self.assertEqual(steps(12), 9)
|
|
|
|
def test_large_number_of_even_and_odd_steps(self):
|
|
self.assertEqual(steps(1000000), 152)
|
|
|
|
def test_zero_is_an_error(self):
|
|
with self.assertRaisesWithMessage(ValueError):
|
|
steps(0)
|
|
|
|
def test_negative_value_is_an_error(self):
|
|
with self.assertRaisesWithMessage(ValueError):
|
|
steps(-15)
|
|
|
|
# Utility functions
|
|
def setUp(self):
|
|
try:
|
|
self.assertRaisesRegex
|
|
except AttributeError:
|
|
self.assertRaisesRegex = self.assertRaisesRegexp
|
|
|
|
def assertRaisesWithMessage(self, exception):
|
|
return self.assertRaisesRegex(exception, r".+")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|