test_lsdi.py 文件源码

def test_data_files(self, lsdibag):
        with patch.object(lsdibag, 'image_files') as mock_imgfiles:
            with patch.object(lsdibag, 'page_text_files') as mock_txtfiles:

                mock_imgfiles.return_value = [
                    '001.tif', '002.tif', '003.tif'
                ]
                mock_txtfiles.return_value = [
                    '001.txt', '002.txt', '003.txt'
                    '001.pos', '002.pos', '003.pos'
                ]

                datafiles = lsdibag.data_files()

                # should include pdf and ocr file
                assert lsdibag.item.pdf in datafiles
                assert lsdibag.item.ocr_file in datafiles
                # should all image and text files
                for imgfile in mock_imgfiles.return_value:
                    assert imgfile in datafiles
                for txtfile in mock_txtfiles.return_value:
                    assert txtfile in datafiles