def test_empty_metadata(self): empty_metadata = pd.DataFrame() act = metadata_utils.list_all_futures_codes(empty_metadata) exp = [] self.assertListEqual(act, exp)
def test(self): act = metadata_utils.list_all_futures_codes(self.metadata) exp = list(set(map(lambda x : x[:-5], self.metadata['contract']))) self.assertListEqual(act, exp)