def test_ingest(self):
"PNG image is ingested and document is created"
inbox_img = self.inbox / 'x.png'
docs_img = self.docs / 'x.png'
shutil.copy(str(TESTDATA / 'shapes.png'), str(inbox_img))
# Read the image for comparison test
original_image = cv2.imread(str(inbox_img))
main([str(self.inbox), str(self.docs)])
# Document, scan and thumbnail should all exists
self.assertTrue((self.docs / 'x.inselect').is_file())
self.assertTrue(docs_img.is_file())
self.assertTrue((self.docs / 'x_thumbnail.jpg').is_file())
# Scan should have been removed from inbox
self.assertFalse(inbox_img.is_file())
# Scan is as expected?
doc = InselectDocument.load(self.docs / 'x.inselect')
self.assertTrue(np.all(original_image == doc.scanned.array))
self.assertTrue(doc.thumbnail.available)
self.assertEqual(4096, doc.thumbnail.array.shape[1])
def test_ingest(self):
"PNG image is ingested and document is created"
inbox_img = self.inbox / 'x.png'
docs_img = self.docs / 'x.png'
shutil.copy(str(TESTDATA / 'shapes.png'), str(inbox_img))
# Read the image for comparison test
original_image = cv2.imread(str(inbox_img))
main([str(self.inbox), str(self.docs)])
# Document, scan and thumbnail should all exists
self.assertTrue((self.docs / 'x.inselect').is_file())
self.assertTrue(docs_img.is_file())
self.assertTrue((self.docs / 'x_thumbnail.jpg').is_file())
# Scan should have been removed from inbox
self.assertFalse(inbox_img.is_file())
# Scan is as expected?
doc = InselectDocument.load(self.docs / 'x.inselect')
self.assertTrue(np.all(original_image == doc.scanned.array))
self.assertTrue(doc.thumbnail.available)
self.assertEqual(4096, doc.thumbnail.array.shape[1])
def test_ingest_create_docs(self):
"Document dir is created on ingest"
docs = self.docs / 'I do not yet exist'
self.assertFalse(docs.is_dir())
# Create an image to ingest
inbox_img = self.inbox / 'x.png'
shutil.copy(str(TESTDATA / 'test_segment.png'), str(inbox_img))
main([unicode(self.inbox), unicode(docs)])
self.assertTrue(docs.is_dir())
def test_cookie_cutter(self):
"Ingested image with cookie cutter applied"
shutil.copy(unicode(TESTDATA / 'test_segment.png'),
unicode(self.inbox / 'x.png'))
main([
'--cookie-cutter={0}'.format(TESTDATA / '2x2.inselect_cookie_cutter'),
unicode(self.inbox), unicode(self.docs)
])
doc = InselectDocument.load(self.docs / 'x.inselect')
self.assertEqual(4, len(doc.items))
def test_ingest_create_docs(self):
"Document dir is created on ingest"
docs = self.docs / 'I do not yet exist'
self.assertFalse(docs.is_dir())
# Create an image to ingest
inbox_img = self.inbox / 'x.png'
shutil.copy(str(TESTDATA / 'shapes.png'), str(inbox_img))
main([str(self.inbox), str(docs)])
self.assertTrue(docs.is_dir())
def test_cookie_cutter(self):
"Ingested image with cookie cutter applied"
shutil.copy(str(TESTDATA / 'shapes.png'), str(self.inbox / 'x.png'))
main([
'--cookie-cutter={0}'.format(TESTDATA /
'2x2.inselect_cookie_cutter'),
str(self.inbox),
str(self.docs)
])
doc = InselectDocument.load(self.docs / 'x.inselect')
self.assertEqual(4, len(doc.items))
def test_extension_cases(self):
"Ingestion of image files with extensions in various combinatons of case"
# Create images to ingest
lower = 'lower.png'
upper = 'upper.png'
title = 'title.Png'
shutil.copy(str(TESTDATA / 'test_segment.png'), str(self.inbox / lower))
shutil.copy(str(TESTDATA / 'test_segment.png'), str(self.inbox / upper))
shutil.copy(str(TESTDATA / 'test_segment.png'), str(self.inbox / title))
main([unicode(self.inbox), unicode(self.docs)])
# Images should have been removed from inbox
self.assertFalse((self.inbox / lower).is_file())
self.assertFalse((self.inbox / upper).is_file())
self.assertFalse((self.inbox / title).is_file())
# Images should have been moved to docs dir
self.assertTrue((self.docs / lower).is_file())
self.assertTrue((self.docs / upper).is_file())
self.assertTrue((self.docs / title).is_file())
def test_extension_cases(self):
"Ingestion of image files with extensions in various combinatons of case"
# Create images to ingest
lower = 'lower.png'
upper = 'upper.png'
title = 'title.Png'
shutil.copy(str(TESTDATA / 'shapes.png'), str(self.inbox / lower))
shutil.copy(str(TESTDATA / 'shapes.png'), str(self.inbox / upper))
shutil.copy(str(TESTDATA / 'shapes.png'), str(self.inbox / title))
main([str(self.inbox), str(self.docs)])
# Images should have been removed from inbox
self.assertFalse((self.inbox / lower).is_file())
self.assertFalse((self.inbox / upper).is_file())
self.assertFalse((self.inbox / title).is_file())
# Images should have been moved to docs dir
self.assertTrue((self.docs / lower).is_file())
self.assertTrue((self.docs / upper).is_file())
self.assertTrue((self.docs / title).is_file())
