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

fixed some sql engine test

parent b48320d2
......@@ -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)
def test_sql_engine_fetch_factor_range(self):
......@@ -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]
np.testing.assert_array_almost_equal(calculated_factor.ROE.values, df.ROE.values)
......@@ -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]
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