image-sort/tests.py at master · sprutner/image-sort · GitHub

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
47
48
49
50
51
import unittest
import tempfile
import os

import sizzle

# Constants
test_folder_path = tempfile.mkdtemp()
test_output_path = os.path.join(test_folder_path, 'output')
test_files = [
    '33.jpg',
    '23_12.jpg',
    'haunted cabinet 2.jpg',
    'spooky.jpg',
    '23-11-332.jpg',
    'test-table-1'
]
expected_dirs = [
    '33',
    '23',
    'spooky',
    'haunted_cabinet',
    'test_table'
]

expected_file_paths = [
    '33/33/jpg',
    '23/23_12.jpg',
    '23_11/23_11_332.jpg',
    'haunted_cabinet/haunted_cabinet_2.jpg',
    'spooky/spooky.jpg',
    'test_table/test_table_1.jpg'
]

class TestSizzle(unittest.TestCase):
    def setUp(self):
        # Creates some test files
        for file in test_files:
            open(os.path.join(test_folder_path, file), 'a').close()

    def test_sizzle(self):
        os.chdir(test_folder_path)
        sizzle.set_testing_globals(test_folder_path)
        # print("test folder path {}".format(test_folder_path))
        sizzle.main()
        # print(os.listdir(test_output_path))

        for dir in expected_dirs:
            # print(os.path.join(test_output_path, dir))
            # print(os.path.isdir(os.path.join(test_output_path, dir)))
            assert os.path.isdir(os.path.join(test_output_path, dir)) == True