diff --git a/tests/test_sources.py b/tests/test_sources.py index 1257ac55..837ee459 100644 --- a/tests/test_sources.py +++ b/tests/test_sources.py @@ -25,11 +25,9 @@ class TestDataFrameSource(TestCase): for expected_dt, expected_price in df.iterrows(): sid0 = source.next() - sid1 = source.next() - assert expected_dt == sid0.dt == sid1.dt + assert expected_dt == sid0.dt assert expected_price[0] == sid0.price - assert expected_price[1] == sid1.price def test_sid_filtering(self): _, df = factory.create_test_df_source()