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'))