def test_insert_no_import_images():
"""Unit test for method insert, no import images case."""
image_database = ImageDatabase(
data_dir=TEST_TEMP_DIRECTORY,
import_images=False
)
image_database.insert(20, 'tests/test_image.jpg')
assert not exists(f'{TEST_TEMP_DIRECTORY}/0/20.jpg') and \
exists(f'{TEST_TEMP_DIRECTORY}/0/20.json')
def test_insert_import_images():
"""Unit test for method insert, import images case."""
image_database = ImageDatabase(
data_dir=TEST_TEMP_DIRECTORY,
import_images=True
)
image_database.insert(10, 'tests/test_image.jpg')
assert exists(f'{TEST_TEMP_DIRECTORY}/0/10.jpg') \
and exists(f'{TEST_TEMP_DIRECTORY}/0/10.json')