From 8ceff892c595e3c76969e6c41fe2353cb7bb3f71 Mon Sep 17 00:00:00 2001 From: Maya Tydykov Date: Thu, 18 Feb 2016 15:54:35 -0500 Subject: [PATCH] BUG: specify UTC and no name-checking. DOC: fix doc. MAINT: revert to using getitem. --- tests/pipeline/test_buyback_auth.py | 6 +++--- tests/pipeline/test_earnings.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/pipeline/test_buyback_auth.py b/tests/pipeline/test_buyback_auth.py index 5d77aec1..6c023e9c 100644 --- a/tests/pipeline/test_buyback_auth.py +++ b/tests/pipeline/test_buyback_auth.py @@ -1,5 +1,5 @@ """ -Tests for the reference loader for EarningsCalendar. +Tests for the reference loader for Buyback Authorizations. """ from functools import partial from unittest import TestCase @@ -103,7 +103,7 @@ param_dates = list(gen_calendars( '2014-01-10', '2014-01-15', '2014-01-20', - ]), + ], utc=True), )) @@ -253,7 +253,7 @@ class BuybackAuthLoaderCommonTest(object): for col_name in self.cols.keys(): assert_series_equal(result[col_name].xs(sid, level=1), self.cols[col_name][sid], - sid) + check_names=False) class CashBuybackAuthLoaderTestCase(TestCase, BuybackAuthLoaderCommonTest): diff --git a/tests/pipeline/test_earnings.py b/tests/pipeline/test_earnings.py index f48251da..185801ba 100644 --- a/tests/pipeline/test_earnings.py +++ b/tests/pipeline/test_earnings.py @@ -392,11 +392,11 @@ class EarningsCalendarLoaderInferTimestampTestCase(TestCase): name=ANNOUNCEMENT_FIELD_NAME), ) assert_series_equal( - loader.events_by_sid[1].loc[:, ANNOUNCEMENT_FIELD_NAME], + loader.events_by_sid[1][ANNOUNCEMENT_FIELD_NAME], pd.Series( - index=announcement_dates[1].loc[:, TS_FIELD_NAME], + index=announcement_dates[1][TS_FIELD_NAME], data=np.array( - announcement_dates[1].loc[:, ANNOUNCEMENT_FIELD_NAME] + announcement_dates[1][ANNOUNCEMENT_FIELD_NAME] ), name=ANNOUNCEMENT_FIELD_NAME )