def test(self): new_metadata = metadata_utils.add_contract_names(self.metadata) row = new_metadata.iloc[0] act = row['contract'] exp = row['code'].split('/')[1] self.assertEqual(act, exp) # Check that we didn't mutate the original metadata act = self.metadata.columns.tolist() exp = ['code','name'] self.assertListEqual(act, exp)
def setUp(self): self.metadata = pd.read_csv(TEST_DATA_DIR + '/dataset-codes.csv') self.metadata = metadata_utils.add_contract_names(self.metadata)