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)