add unittest for random_expression

test/test_random_expression.py

@@ -0,0 +1,100 @@
+#!/usr/bin/env python
+# encoding: utf-8
+
+
+import unittest
+
+from pymath.random_expression import RdExpression
+
+
+class TestRandomExpression(unittest.TestCase):
+    """Testing functions from pymath.random_expression"""
+
+    def test_only_form(self):
+        form = "{a} + 2"
+        rdExp = RdExpression(form)
+
+        self.assertEqual(rdExp._letters, {'a'})
+        self.assertEqual(rdExp._2replaced, {'a'})
+
+        rdExp()
+        self.assertEqual(set(rdExp._gene_varia.keys()), {'a'}) 
+        self.assertEqual(set(rdExp._gene_2replaced.keys()), {'a'}) 
+
+    def test_only_form_calc(self):
+        form = "{a + b} + 2"
+        rdExp = RdExpression(form)
+
+        self.assertEqual(rdExp._letters, {'a', 'b'})
+        self.assertEqual(rdExp._2replaced, {'a + b'})
+
+        rdExp()
+        self.assertEqual(set(rdExp._gene_varia.keys()), {'a', 'b'}) 
+        self.assertEqual(set(rdExp._gene_2replaced.keys()), {'a + b'}) 
+
+    def test_only_form_cond(self):
+        form = "{a} + 2"
+        cond = ["{a} == 3"]
+        rdExp = RdExpression(form, cond)
+
+        self.assertEqual(rdExp._letters, {'a'})
+        self.assertEqual(rdExp._2replaced, {'a'})
+
+        rdExp()
+        self.assertEqual(set(rdExp._gene_varia.keys()), {'a'}) 
+        self.assertEqual(set(rdExp._gene_2replaced.keys()), {'a'}) 
+
+        self.assertEqual(rdExp._gene_varia['a'], 3)
+
+    def test_only_form_conds(self):
+        form = "{a} + 2"
+        cond = ["{a} in list(range(5))", "{a} % 2 == 1"]
+        rdExp = RdExpression(form, cond)
+
+        self.assertEqual(rdExp._letters, {'a'})
+        self.assertEqual(rdExp._2replaced, {'a'})
+
+        rdExp()
+        self.assertEqual(set(rdExp._gene_varia.keys()), {'a'}) 
+        self.assertEqual(set(rdExp._gene_2replaced.keys()), {'a'}) 
+
+        self.assertTrue(rdExp._gene_varia['a'] in list(range(5)))
+        self.assertTrue(rdExp._gene_varia['a'] % 2 == 1)
+
+    def test_only_form_calc_cond(self):
+        form = "{a*3} * {b}"
+        cond = ["{a} == 3"]
+        rdExp = RdExpression(form, cond)
+
+        self.assertEqual(rdExp._letters, {'a', 'b'})
+        self.assertEqual(rdExp._2replaced, {'a', 'b', 'a*3'})
+
+        rdExp()
+        self.assertEqual(set(rdExp._gene_varia.keys()), {'a', 'b'}) 
+        self.assertEqual(set(rdExp._gene_2replaced.keys()), {'a', 'b', 'a*3'}) 
+
+        self.assertEqual(rdExp._gene_varia['a'], 3)
+
+
+    def test_only_form_calc_cond_calc(self):
+        form = "{a} + 2"
+
+        pass
+
+
+
+if __name__ == '__main__':
+    unittest.main()
+
+
+
+
+
+
+
+
+# -----------------------------
+# Reglages pour 'vim'
+# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
+# cursor: 16 del 
+