added tests for universe

alphamind/tests/data/engines/test_universe.py

+# -*- coding: utf-8 -*-
+"""
+Created on 2018-2-9
+
+@author: cheng.li
+"""
+
+import unittest
+from PyFin.api import LAST
+from alphamind.data.engines.universe import Universe
+
+
+class TestUniverse(unittest.TestCase):
+
+    def test_universe_persistence(self):
+        universe = Universe('custom', ['zz500'])
+        univ_desc = universe.save()
+        loaded_universe = Universe.load(univ_desc)
+
+        self.assertEqual(universe.name, loaded_universe.name)
+        self.assertListEqual(universe.base_universe, loaded_universe.base_universe)
+
+        universe = Universe('custom', ['zz500'], filter_cond=LAST('x') > 1.)
+        univ_desc = universe.save()
+        loaded_universe = Universe.load(univ_desc)
+
+        self.assertEqual(universe.name, loaded_universe.name)
+        self.assertListEqual(universe.base_universe, loaded_universe.base_universe)
+        self.assertEqual(universe.filter_cond, loaded_universe.filter_cond)
\ No newline at end of file

alphamind/tests/test_suite.py

@@ -15,6 +15,7 @@ from alphamind.tests.data.test_neutralize import TestNeutralize
 from alphamind.tests.data.test_standardize import TestStandardize
 from alphamind.tests.data.test_winsorize import TestWinsorize
 from alphamind.tests.data.test_quantile import TestQuantile
+from alphamind.tests.data.engines.test_universe import TestUniverse
 from alphamind.tests.portfolio.test_constraints import TestConstraints
 from alphamind.tests.portfolio.test_evolver import TestEvolver
 from alphamind.tests.portfolio.test_longshortbuild import TestLongShortBuild
@@ -43,6 +44,7 @@ if __name__ == '__main__':
                          TestStandardize,
                          TestWinsorize,
                          TestQuantile,
+                         TestUniverse,
                          TestConstraints,
                          TestEvolver,
                          TestLongShortBuild,