aboutsummaryrefslogtreecommitdiff
path: root/backend/tests/test_review_imgs_to_gen.py
diff options
context:
space:
mode:
authorTerry Truong <terry06890@gmail.com>2023-01-29 11:30:47 +1100
committerTerry Truong <terry06890@gmail.com>2023-01-29 11:30:47 +1100
commit8781fdb2b8c530a6c1531ae9e82221eb062e34fb (patch)
treeffd824aa9b945d69b47f012617ee13d98764d078 /backend/tests/test_review_imgs_to_gen.py
parentf5e87ae628bab0eef97b3e3e62f6d71cca9c99c0 (diff)
Adjust backend coding style
Add line spacing, section comments, and import consistency
Diffstat (limited to 'backend/tests/test_review_imgs_to_gen.py')
-rw-r--r--backend/tests/test_review_imgs_to_gen.py9
1 files changed, 8 insertions, 1 deletions
diff --git a/backend/tests/test_review_imgs_to_gen.py b/backend/tests/test_review_imgs_to_gen.py
index d88523b..e98ab32 100644
--- a/backend/tests/test_review_imgs_to_gen.py
+++ b/backend/tests/test_review_imgs_to_gen.py
@@ -1,5 +1,7 @@
import unittest
-import tempfile, os, shutil
+import tempfile
+import os
+import shutil
from tests.common import readTestFile, createTestDbTable
from tol_data.review_imgs_to_gen import reviewImgs
@@ -62,19 +64,24 @@ class TestReviewImgs(unittest.TestCase):
('four', 4),
}
)
+
# Run
outFile = os.path.join(tempDir, 'imgList.txt')
reviewImgs(eolImgDir, enwikiImgDir, dbFile, outFile, 'all')
+
# Check
self.assertEqual(set(readTestFile(outFile).splitlines()), {
'ott1 ' + os.path.join(eolImgDir, '1 10.jpg'),
'ott2',
'ott3 ' + os.path.join(enwikiImgDir, '3.png'),
})
+
# Add extra data
createTestDbTable(dbFile, None, 'INSERT INTO nodes VALUES (?, ?, ?)',{('four', 'ott4', 2)})
+
# Run
reviewImgs(eolImgDir, enwikiImgDir, dbFile, outFile, 'all')
+
# Check
self.assertEqual(set(readTestFile(outFile).splitlines()), {
'ott1 ' + os.path.join(eolImgDir, '1 10.jpg'),