Commit 4df5acb4 authored by Dr.李's avatar Dr.李

fixed some sql engine test

parent b48320d2
...@@ -198,7 +198,7 @@ class TestSqlEngine(unittest.TestCase): ...@@ -198,7 +198,7 @@ class TestSqlEngine(unittest.TestCase):
) )
) )
df = pd.read_sql(query, con=self.engine.engine).sort_values('code').dropna() df = pd.read_sql(query, con=self.engine.engine).sort_values('code')
np.testing.assert_array_almost_equal(factor_data.ROE.values, df.ROE.values) np.testing.assert_array_almost_equal(factor_data.ROE.values, df.ROE.values)
def test_sql_engine_fetch_factor_range(self): def test_sql_engine_fetch_factor_range(self):
...@@ -219,7 +219,7 @@ class TestSqlEngine(unittest.TestCase): ...@@ -219,7 +219,7 @@ class TestSqlEngine(unittest.TestCase):
) )
) )
df = pd.read_sql(query, con=self.engine.engine).dropna() df = pd.read_sql(query, con=self.engine.engine)
calculated_factor = factor_data[factor_data.trade_date == ref_date] calculated_factor = factor_data[factor_data.trade_date == ref_date]
np.testing.assert_array_almost_equal(calculated_factor.ROE.values, df.ROE.values) np.testing.assert_array_almost_equal(calculated_factor.ROE.values, df.ROE.values)
...@@ -243,7 +243,7 @@ class TestSqlEngine(unittest.TestCase): ...@@ -243,7 +243,7 @@ class TestSqlEngine(unittest.TestCase):
) )
) )
df = pd.read_sql(query, con=self.engine.engine).dropna() df = pd.read_sql(query, con=self.engine.engine)
calculated_factor = factor_data[factor_data.trade_date == ref_date] calculated_factor = factor_data[factor_data.trade_date == ref_date]
np.testing.assert_array_almost_equal(calculated_factor.dx.values, df.ROE.values) np.testing.assert_array_almost_equal(calculated_factor.dx.values, df.ROE.values)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment