From 2dbf905d43019e40943dd10ca8fdcf8e3648cb28 Mon Sep 17 00:00:00 2001 From: Thomas Wiecki Date: Wed, 19 Sep 2012 20:01:10 -0400 Subject: [PATCH] Added unittest for dataframe source. --- tests/test_sources.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 tests/test_sources.py diff --git a/tests/test_sources.py b/tests/test_sources.py new file mode 100644 index 00000000..0711f72f --- /dev/null +++ b/tests/test_sources.py @@ -0,0 +1,13 @@ +import zipline.utils.factory as factory + +def test_dataframe_source(): + source, df = factory.create_test_df_source() + + for expected_dt, expected_price in df.iterrows(): + sid0 = source.next() + sid1 = source.next() + + assert expected_dt == sid0.dt == sid1.dt + assert expected_price[0] == sid0.price + assert expected_price[1] == sid1.price +