update tests

alphamind/tests/data/engines/test_sql_engine.py

@@ -424,7 +424,6 @@ class TestSqlEngine(unittest.TestCase):
         df3 = self.engine.fetch_factor(ref_date, raw_factor, codes)
 
         ind_matrix = self.engine.fetch_industry_matrix(ref_date, codes, 'sw', 1)
-
         cols = sorted(ind_matrix.columns[2:].tolist())
 
         series = (ind_matrix[cols] * np.array(range(1, len(cols)+1))).sum(axis=1)

alphamind/tests/data/engines/test_universe.py

@@ -25,5 +25,20 @@ class TestUniverse(unittest.TestCase):
         universe = Universe('zz500')
         univ_desc = universe.save()
         loaded_universe = load_universe(univ_desc)
+        self.assertEqual(universe, loaded_universe)
 
+    def test_universe_arithmic(self):
+        universe = Universe('zz500') + Universe('hs300')
+        univ_desc = universe.save()
+        loaded_universe = load_universe(univ_desc)
+        self.assertEqual(universe, loaded_universe)
+
+        universe = Universe('zz500') - Universe('hs300')
+        univ_desc = universe.save()
+        loaded_universe = load_universe(univ_desc)
+        self.assertEqual(universe, loaded_universe)
+
+        universe = Universe('zz500') & Universe('hs300')
+        univ_desc = universe.save()
+        loaded_universe = load_universe(univ_desc)
         self.assertEqual(universe, loaded_universe)
\ No newline at end of file