def test_crop(self): with copy_of_directory(assets.path_to('dfki-testdata/data')) as wsdir: ws = Workspace(self.resolver, wsdir) pagexml_before = len(ws.mets.find_files(mimetype=MIMETYPE_PAGE)) run_processor( OcrdAnybaseocrCropper, resolver=self.resolver, mets_url=str(Path(wsdir, 'mets.xml')), input_file_grp='BIN', output_file_grp='CROP-TEST', parameter={}, ) ws.reload_mets() pagexml_after = len(ws.mets.find_files(mimetype=MIMETYPE_PAGE)) self.assertEqual(pagexml_after, pagexml_before + 1)
def test_crop(self): if not torch.cuda.is_available(): pytest.skip('CUDA is not available, cannot test dewarping') with copy_of_directory(assets.path_to('dfki-testdata/data')) as wsdir: ws = Workspace(self.resolver, wsdir) pagexml_before = len(ws.mets.find_files(mimetype=MIMETYPE_PAGE)) run_processor(OcrdAnybaseocrDewarper, resolver=self.resolver, mets_url=str(Path(wsdir, 'mets.xml')), input_file_grp='BIN', output_file_grp='DEWARP-TEST', parameter={'model_path': str(self.model_path)}) ws.reload_mets() pagexml_after = len(ws.mets.find_files(mimetype=MIMETYPE_PAGE)) self.assertEqual(pagexml_after, pagexml_before + 1)