<filename>tests/test_software_loader.py
from maskgen import software_loader
import unittest


class TestSoftwareLoader(unittest.TestCase):

    def test_load(self):
        ops = software_loader.getOperationsByCategory('image', 'image')
        self.assertTrue('Mosaic' in ops['AdditionalEffect'])
        self.assertTrue('AntiForensicExifQuantizationTable' in ops['AntiForensic'])
        self.assertFalse('FilterColorLUT' in ops['Filter'])
        self.assertFalse('PasteImageSpliceToFrames' in ops['Paste'])
        ops = software_loader.getOperationsByCategory('video', 'video')
        self.assertTrue('PasteImageSpliceToFrames' in ops['Paste'])

    def test_project_properties(self):
        name = "Custom"
        values = ["Custom 0", "Custom 1", "Custom 10", "Custom 11"]
        prop = software_loader.ProjectProperty(name=name, values=software_loader.Operation(name=name)
                                               .getParameterValuesForType(name, 'image', values))

        self.assertTrue(prop.values == values)


if __name__ == '__main__':
    unittest.main()

<filename>tests/test_software_loader.py
from maskgen import software_loader
import unittest


class TestSoftwareLoader(unittest.TestCase):

    def test_load(self):
        ops = software_loader.getOperationsByCategory('image', 'image')
        self.assertTrue('Mosaic' in ops['AdditionalEffect'])
        self.assertTrue('AntiForensicExifQuantizationTable' in ops['AntiForensic'])
        self.assertFalse('FilterColorLUT' in ops['Filter'])
        self.assertFalse('PasteImageSpliceToFrames' in ops['Paste'])
        ops = software_loader.getOperationsByCategory('video', 'video')
        self.assertTrue('PasteImageSpliceToFrames' in ops['Paste'])

    def test_project_properties(self):
        name = "Custom"
        values = ["Custom 0", "Custom 1", "Custom 10", "Custom 11"]
        prop = software_loader.ProjectProperty(name=name, values=software_loader.Operation(name=name)
                                               .getParameterValuesForType(name, 'image', values))

        self.assertTrue(prop.values == values)


if __name__ == '__main__':
    unittest.main()
