def test_with_scalar(self): df = dataAccess._remove_storms_from_df(self.df, self.exclude_dates[0], 'datecol1') known_shape = (18, 3) nt.assert_tuple_equal(df.shape, known_shape)
def test_datecol2(self): df = dataAccess._remove_storms_from_df(self.df, self.exclude_dates, 'datecol2') nt.assert_tuple_equal(df.shape, self.known_datecol2_shape)