def test_transform_already_path(self, gt_dataframe): data = gt_dataframe.copy() data['path'] = 'a path' transformer = PathGetter(env='prd') with pytest.raises(RuntimeError): transformer.fit(data) data = gt_dataframe.copy() transformer.fit(data) data['path'] = 'a path' with pytest.raises(RuntimeError): transformer.transform(data)
def test_transform(self, gt_dataframe, ts_dataframe): ground_truth_path = os.path.join('..', 'ground_truth') train_set_path = os.path.join('..', 'dumps', 'prd') transformer = PathGetter(env='prd', ground_truth_uids=gt_dataframe.index, ground_truth_path=ground_truth_path, train_set_path=train_set_path) data_dataframe = pd.concat([gt_dataframe, ts_dataframe], axis=0) transformer.fit(data_dataframe) transformed = transformer.transform(data_dataframe) gt_uid = data_dataframe.loc[data_dataframe['col'] == 'GT'].index[0] ts_uid = data_dataframe.loc[data_dataframe['col'] == 'TS'].index[0] assert (transformed.loc[gt_uid, 'path'] == os.path.join( ground_truth_path, gt_uid, 'FTF.pdf')) assert (transformed.loc[ts_uid, 'path'] == os.path.join( train_set_path, ts_uid, 'FTF.pdf'))