1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
import unittest
import tempfile, os, shutil
from tests.common import createTestDbTable
from tol_data.eol.review_imgs import reviewImgs
CLICK_IMG = os.path.join(os.path.dirname(__file__), '..', 'green.png')
AVOID_IMG = os.path.join(os.path.dirname(__file__), '..', 'red.png')
class TestReviewImgs(unittest.TestCase):
def test_review(self):
with tempfile.TemporaryDirectory() as tempDir:
# Create input images
imgDir = os.path.join(tempDir, 'imgs_for_review')
os.mkdir(imgDir)
shutil.copy(CLICK_IMG, os.path.join(imgDir, '1 10.jpg'))
shutil.copy(CLICK_IMG, os.path.join(imgDir, '2 20.jpeg'))
shutil.copy(AVOID_IMG, os.path.join(imgDir, '2 21.gif'))
shutil.copy(AVOID_IMG, os.path.join(imgDir, '2 22.jpg'))
shutil.copy(AVOID_IMG, os.path.join(imgDir, '3 30.png'))
shutil.copy(AVOID_IMG, os.path.join(imgDir, '3 31.jpg'))
# Create temp extra-info db
extraInfoDb = os.path.join(tempDir, 'data.db')
createTestDbTable(
extraInfoDb,
'CREATE TABLE eol_ids (name TEXT PRIMARY KEY, id INT)',
'INSERT INTO eol_ids VALUES (?, ?)',
{
('one', 1),
('two', 2),
('three', 3),
}
)
createTestDbTable(
extraInfoDb,
'CREATE TABLE names(name TEXT, alt_name TEXT, pref_alt INT, src TEXT, PRIMARY KEY(name, alt_name))',
'INSERT OR IGNORE INTO names VALUES (?, ?, ?, ?)',
{
('two','II',1,'eol'),
}
)
# Run
outDir = os.path.join(tempDir, 'imgs')
reviewImgs(imgDir, outDir, extraInfoDb)
# Check
self.assertEqual(set(os.listdir(outDir)), {'1 10.jpg', '2 20.jpeg'})
|
