diff --git a/main.py b/main.py index aecd51e..1488c24 100644 --- a/main.py +++ b/main.py @@ -1,3 +1,6 @@ +def divide_by_one(num): # Is whole number + return float(num).is_integer() + def divide_by_two(num): # Take last digit, divide by two only on that return _get_last_digit(num) % 2 == 0 diff --git a/tests/one_test.py b/tests/one_test.py new file mode 100644 index 0000000..35035ae --- /dev/null +++ b/tests/one_test.py @@ -0,0 +1,12 @@ +from main import divide_by_one +import unittest + +class TestDivideByOne(unittest.TestCase): + def test_by_1fractional(self): + self.assertFalse(divide_by_one(1.5)) + def test_by_1(self): + self.assertTrue(divide_by_one(1)) + def test_by_2(self): + self.assertTrue(divide_by_one(2)) + def test_by_362880(self): + self.assertTrue(divide_by_one(362880))