From 1ae6037a813bc1d97832446d1dbaa340e4ff3425 Mon Sep 17 00:00:00 2001 From: Andrew Daniels Date: Wed, 24 Jun 2015 13:03:14 -0400 Subject: [PATCH] TST: Correct and clean up mock futures data for clarity --- tests/test_assets.py | 23 +++++++++++++---------- 1 file changed, 13 insertions(+), 10 deletions(-) diff --git a/tests/test_assets.py b/tests/test_assets.py index f05f5bc7..994e6cbe 100644 --- a/tests/test_assets.py +++ b/tests/test_assets.py @@ -227,24 +227,27 @@ class AssetTestCase(TestCase): class TestFuture(TestCase): - - future = Future(2468, - symbol='OMK15', - notice_date='2014-01-20', - expiration_date='2014-02-20', - contract_multiplier=500) + future = Future( + 2468, + symbol='OMH15', + notice_date=pd.Timestamp('2014-01-20', tz='UTC'), + expiration_date=pd.Timestamp('2014-02-20', tz='UTC'), + contract_multiplier=500 + ) def test_str(self): strd = self.future.__str__() - self.assertEqual("Future(2468 [OMK15])", strd) + self.assertEqual("Future(2468 [OMH15])", strd) def test_repr(self): reprd = self.future.__repr__() self.assertTrue("Future" in reprd) self.assertTrue("2468" in reprd) - self.assertTrue("OMK15" in reprd) - self.assertTrue("notice_date='2014-01-20'" in reprd) - self.assertTrue("expiration_date='2014-02-20'" in reprd) + self.assertTrue("OMH15" in reprd) + self.assertTrue(("notice_date=Timestamp('2014-01-20 00:00:00+0000', " + "tz='UTC')") in reprd) + self.assertTrue("expiration_date=Timestamp('2014-02-20 00:00:00+0000'" + in reprd) self.assertTrue("contract_multiplier=500" in reprd) def test_reduce(self):