|
@@ -1,5 +1,5 @@
|
|
|
from src.node import ExpressionNode as N
|
|
from src.node import ExpressionNode as N
|
|
|
-from src.rules.utils import nary_node, is_prime, least_common_multiple
|
|
|
|
|
|
|
+from src.rules.utils import nary_node, least_common_multiple
|
|
|
from tests.rulestestcase import RulesTestCase, tree
|
|
from tests.rulestestcase import RulesTestCase, tree
|
|
|
|
|
|
|
|
|
|
|
|
@@ -15,17 +15,6 @@ class TestRulesUtils(RulesTestCase):
|
|
|
self.assertEqualNodes(nary_node('+', [a, b, c, d]),
|
|
self.assertEqualNodes(nary_node('+', [a, b, c, d]),
|
|
|
N('+', N('+', N('+', a, b), c), d))
|
|
N('+', N('+', N('+', a, b), c), d))
|
|
|
|
|
|
|
|
- def test_is_prime(self):
|
|
|
|
|
- self.assertTrue(is_prime(2))
|
|
|
|
|
- self.assertTrue(is_prime(3))
|
|
|
|
|
- self.assertTrue(is_prime(647))
|
|
|
|
|
-
|
|
|
|
|
- self.assertFalse(is_prime(0))
|
|
|
|
|
- self.assertFalse(is_prime(1))
|
|
|
|
|
- self.assertFalse(is_prime(4))
|
|
|
|
|
- self.assertFalse(is_prime(9))
|
|
|
|
|
- self.assertFalse(is_prime(100))
|
|
|
|
|
-
|
|
|
|
|
def test_least_common_multiple(self):
|
|
def test_least_common_multiple(self):
|
|
|
self.assertEqual(least_common_multiple(5, 6), 30)
|
|
self.assertEqual(least_common_multiple(5, 6), 30)
|
|
|
self.assertEqual(least_common_multiple(5, 6, 15), 30)
|
|
self.assertEqual(least_common_multiple(5, 6, 15), 30)
|