Test suite and test runner added

tests/README.md

@@ -0,0 +1,8 @@
+# Tests
+
+To run all tests simply run the following command from the root directory:
+
+```python
+>>> python tests/run.py
+```
+

tests/run.py

@@ -0,0 +1,23 @@
+import unittest
+import test_ga
+import test_lm
+import test_ps
+import test_utilities
+
+
+def all_suites():
+
+    suites = [
+        test_ga.suite(),
+        test_lm.suite(),
+        test_ps.suite(),
+        test_utilities.suite()
+    ]
+
+    all_suites = unittest.TestSuite(suites)
+    return all_suites
+
+if __name__ == '__main__':
+    runner = unittest.TextTestRunner()
+    test_suite = all_suites()
+    runner.run (test_suite)

tests/test_ga.py

@@ -70,5 +70,13 @@ def test_ga(self):
             next_err = errors[i]
             self.assertGreaterEqual(prev_err, next_err)
 
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(TestGA('test_ga'))
+
+    return suite
+
+
 if __name__ == '__main__':
     unittest.main()

tests/test_lm.py

@@ -86,5 +86,13 @@ def test_lm(self):
         self.lm.save_plots()
         # TODO: Check if all plots are saved
 
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(TestLM('test_lm'))
+
+    return suite
+
+
 if __name__ == '__main__':
     unittest.main()

tests/test_ps.py

@@ -72,5 +72,13 @@ def test_ps(self):
             next_err = errors[i]
             self.assertGreaterEqual(prev_err, next_err)
 
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(TestPS('test_ps'))
+
+    return suite
+
+
 if __name__ == '__main__':
     unittest.main()

tests/test_utilities.py

@@ -12,7 +12,7 @@
 from modestpy.utilities.delete_logs import delete_logs
 
 
-class TestDeleteLogs(unittest.TestCase):
+class TestUtilities(unittest.TestCase):
     def setUp(self):
         # Temp directory
         self.temp_dir = tempfile.mkdtemp()
@@ -24,7 +24,7 @@ def setUp(self):
     def tearDown(self):
         try:
             os.remove(self.log_path)
-        except WindowsError as e:
+        except OSError as e:
             pass # File already removed
         os.rmdir(self.temp_dir)
 
@@ -34,5 +34,12 @@ def test_delete_logs(self):
         self.assertEqual(len(content), 0)
 
 
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(TestUtilities('test_delete_logs'))
+
+    return suite
+
+
 if __name__ == "__main__":
     unittest.main()