From 59dbffb3aab093ff300e79f48f8800c786db683f Mon Sep 17 00:00:00 2001 From: Eddie Hebert Date: Sun, 6 Jan 2013 17:11:38 -0500 Subject: [PATCH] Changes test sources to use unit test's assert instead of plain assert. --- tests/test_sources.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/test_sources.py b/tests/test_sources.py index e316a5d8..4585cdc6 100644 --- a/tests/test_sources.py +++ b/tests/test_sources.py @@ -43,12 +43,12 @@ class TestDataFrameSource(TestCase): assert isinstance(source.start, pd.lib.Timestamp) assert isinstance(source.end, pd.lib.Timestamp) for event in source: - assert 'sid' in event - assert 'arbitrary' in event - assert 'volume' in event - assert 'price' in event - assert event['arbitrary'] == 1. - assert event['volume'] == 1000 - assert event['sid'] == 0 - assert isinstance(event['volume'], int) - assert isinstance(event['arbitrary'], float) + self.assertTrue('sid' in event) + self.assertTrue('arbitrary' in event) + self.assertTrue('volume' in event) + self.assertTrue('price' in event) + self.assertEquals(event['arbitrary'], 1.) + self.assertEquals(event['volume'], 1000) + self.assertEquals(event['sid'], 0) + self.assertTrue(isinstance(event['volume'], int)) + self.assertTrue(isinstance(event['arbitrary'], float))