introduced python tests for mlmc

python/test_experiments.py

+import sys
+from unittest import TestCase, main
+
+sys.path.append('.')
+
+from mlmc_mppy import mpp
+
+
+class TestExperiments(TestCase):
+    @classmethod
+    def setUpClass(cls):
+        mpp.mute = False
+        mpp.build()
+
+    @classmethod
+    def tearDownClass(cls):
+        # mpp.clean_build()
+        pass
+
+    def setUp(self):
+        mpp.clean_data()
+
+    def tearDown(self):
+        mpp.clean_data()
+
+
+class TestBasics(TestExperiments):
+    def test_git_repo(self):
+        self.assertEqual(mpp.check_git_project(), 0)
+
+    def test_build(self):
+        self.assertEqual(mpp.build(), 0)
+
+    def test_run(self):
+        kwargs = {'maxLevel': 5}
+        self.assertEqual(mpp.run(4, config='mlmc_elliptic', kwargs=kwargs), 0)
+
+
+if __name__ == '__main__':
+    main()