36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
# These tests are auto-generated with test data from:
|
|
# https://github.com/exercism/problem-specifications/tree/main/exercises/collatz-conjecture/canonical-data.json
|
|
# File last updated on 2023-07-20
|
|
|
|
import unittest
|
|
|
|
from collatz_conjecture import (
|
|
steps,
|
|
)
|
|
|
|
|
|
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.assertRaises(ValueError) as err:
|
|
steps(0)
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "Only positive integers are allowed")
|
|
|
|
def test_negative_value_is_an_error(self):
|
|
with self.assertRaises(ValueError) as err:
|
|
steps(-15)
|
|
self.assertEqual(type(err.exception), ValueError)
|
|
self.assertEqual(err.exception.args[0], "Only positive integers are allowed")
|