wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-21 12:30:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

TST: Test fixes for pandas update

Browse Source
This commit is contained in:
Richard Frank
2016-02-11 18:46:43 -05:00
parent ff485950b8
commit f14002a741
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/pipeline/test_earnings.py
+7 -1
View File
@@ -260,7 +260,7 @@ class EarningsCalendarLoaderTestCase(TestCase):
'2014-01-10',
'2014-01-15',
'2014-01-20',
]),
], utc=True),
))
def test_compute_earnings(self, dates):
@@ -296,10 +296,12 @@ class EarningsCalendarLoaderTestCase(TestCase):
assert_series_equal(
computed_next.isnull(),
computed_next_busday_offset.isnull(),
check_names=False,
)
assert_series_equal(
computed_previous.isnull(),
computed_previous_busday_offset.isnull(),
check_names=False,
)
for sid in self.sids:
@@ -308,24 +310,28 @@ class EarningsCalendarLoaderTestCase(TestCase):
computed_next.xs(sid, level=1),
expected_next(sid),
sid,
check_names=False,
)
assert_series_equal(
computed_previous.xs(sid, level=1),
expected_previous(sid),
sid,
check_names=False,
)
assert_series_equal(
computed_next_busday_offset.xs(sid, level=1),
expected_next_busday_offset(sid),
sid,
check_names=False,
)
assert_series_equal(
computed_previous_busday_offset.xs(sid, level=1),
expected_previous_busday_offset(sid),
sid,
check_names=False,
)