def test_color_mode_setvalue(self): self.cp['DATASET'] = { "data_entry": "image/data_entry.csv", "class_names": "", "output_dir": "" } self.cp['IMAGE'] = {"image_dir": "image/", "color_mode": "rgb"} ic = ImageConfig(self.cp) self.assertEqual(ic.color_mode, "rgb") self.cp['IMAGE'] = {"image_dir": "image/", "color_mode": "grayscale"} ic = ImageConfig(self.cp) self.assertEqual(ic.color_mode, "grayscale") self.cp['IMAGE'] = {"image_dir": "image/", "color_mode": "hsv"} ic = ImageConfig(self.cp) self.assertEqual(ic.color_mode, "hsv")
def test_image_dimension_setvalue(self): self.cp['DATASET'] = { "data_entry": "image/data_entry.csv", "class_names": "", "output_dir": "" } self.cp['IMAGE'] = {"image_dir": "image/", "image_dimension": 1024} ic = ImageConfig(self.cp) self.assertEqual(ic.img_dim, 1024) self.cp['IMAGE'] = {"image_dir": "image/", "image_dimension": 512} ic = ImageConfig(self.cp) self.assertEqual(ic.img_dim, 512) self.cp['IMAGE'] = {"image_dir": "image/", "image_dimension": 256} ic = ImageConfig(self.cp) self.assertEqual(ic.img_dim, 256)
def test_scale_setvalue(self): self.cp['DATASET'] = { "data_entry": "image/data_entry.csv", "class_names": "", "output_dir": "" } self.cp['IMAGE'] = {"image_dir": "image/", "scale": 1.0} ic = ImageConfig(self.cp) self.assertEqual(ic.scale, 1.0) self.cp['IMAGE'] = {"image_dir": "image/", "scale": 0.5} ic = ImageConfig(self.cp) self.assertEqual(ic.scale, 0.5) self.cp['IMAGE'] = {"image_dir": "image/", "scale": 0.1} ic = ImageConfig(self.cp) self.assertEqual(ic.scale, 0.1)
def test_image_dimension_default(self): self.cp['DATASET'] = { "data_entry": "image/data_entry.csv", "class_names": "", "output_dir": "" } self.cp['IMAGE'] = {"image_dir": "image/"} ic = ImageConfig(self.cp) self.assertEqual(ic.img_dim, 256)
def test_scale_default(self): self.cp['DATASET'] = { "data_entry": "image/data_entry.csv", "class_names": "", "output_dir": "" } self.cp['IMAGE'] = {"image_dir": "image/"} ic = ImageConfig(self.cp) self.assertEqual(ic.scale, 1. / 255)
def test_color_mode_type(self): ic = ImageConfig(self.cp) self.assertIsInstance(ic.color_mode, str)
def test_scale_type(self): ic = ImageConfig(self.cp) self.assertIsInstance(ic.scale, float)
def test_image_dimension_type(self): ic = ImageConfig(self.cp) self.assertIsInstance(ic.img_dim, int)
def test_ImageConfigEqual(self): ic = ImageConfig(self.cp) dc = DatasetConfig(self.cp) self.assertEqual(dc.ImageConfig, ic)
def ImageConfig(self): return ImageConfig(self.cp)