restore test suite

1 files changed, 23 insertions, 0 deletions

diff --git a/test-chill/unit-tests/test_omega.py b/test-chill/unit-tests/test_omega.py
new file mode 100644
index 0000000..91d6a13
--- /dev/null
+++ b/test-chill/unit-tests/test_omega.py
@@ -0,0 +1,23 @@
+import os
+import unittest
+
+import testchill.omega
+import testchill.util
+
+
+class TestOmegaTestCases(unittest.TestCase):
+    def setUp(self):
+        self.omega_dev_dir = os.getenv('OMEGA_DEV_SRC')
+        self.omega_rel_dir = os.getenv('OMEGA_RELEASE_SRC')
+    
+    def tearDown(self):
+        pass
+    
+    def test_omega_dev(self):
+        tc = testchill.omega.BuildOmegaTestCase(self.omega_dev_dir)
+        tc.run()
+        
+    def test_omega_release(self):
+        tc = testchill.omega.BuildOmegaTestCase(self.omega_rel_dir, 'release')
+        tc.run()
+