def setUp(self):
self.working_dir = {}
self.img_model = ImgModel()
self.mask_model = MaskModel()
self.mask_widget = MaskWidget()
self.mask_controller = MaskController(self.working_dir,
self.mask_widget, self.img_model,
self.mask_model)
def setUp(self):
self.app = QtGui.QApplication(sys.argv)
self.working_dir = {}
self.img_model = ImgModel()
self.mask_model = MaskModel()
self.mask_widget = MaskWidget()
self.mask_controller = MaskController(self.working_dir, self.mask_widget, self.img_model, self.mask_model)
class MaskControllerTest(unittest.TestCase):
def setUp(self):
self.app = QtGui.QApplication(sys.argv)
self.working_dir = {}
self.img_model = ImgModel()
self.mask_model = MaskModel()
self.mask_widget = MaskWidget()
self.mask_controller = MaskController(self.working_dir, self.mask_widget, self.img_model, self.mask_model)
def tearDown(self):
del self.img_model
del self.mask_model
self.mask_widget.close()
del self.mask_widget
del self.mask_controller
del self.app
gc.collect()
def get_file_size(self, filename):
stat_info = os.stat(filename)
return stat_info.st_size
def test_loading_and_saving_mask_files(self):
self.mask_controller.load_mask_btn_click(os.path.join(data_path, 'test.mask'))
self.mask_model.mask_below_threshold(self.img_model, 1)
filename = os.path.join(data_path, 'dummy.mask')
self.mask_controller.save_mask_btn_click(filename)
self.assertTrue(os.path.exists(filename))
os.remove(filename)
def test_loading_and_saving_with_super_sampling(self):
self.mask_model.set_supersampling(2)
self.mask_controller.load_mask_btn_click(os.path.join(data_path, 'test.mask'))
self.assertEqual(self.mask_model.get_mask().shape[0], 4096)
self.assertEqual(self.mask_model.get_mask().shape[1], 4096)
self.assertEqual(self.mask_model.get_img().shape[0], 2048)
self.assertEqual(self.mask_model.get_img().shape[1], 2048)
self.mask_model.mask_below_threshold(self.img_model, 1)
filename = os.path.join(data_path, 'dummy.mask')
self.mask_controller.save_mask_btn_click(filename)
self.assertEqual(self.get_file_size(filename), self.get_file_size(os.path.join(data_path, 'test.mask')))
def test_grow_and_shrinking(self):
self.mask_model.mask_ellipse(100,100,20,20)
previous_mask = np.copy(self.mask_model._mask_data)
QTest.mouseClick(self.mask_widget.grow_btn, QtCore.Qt.LeftButton)
self.assertFalse(np.array_equal(previous_mask, self.mask_model._mask_data))
QTest.mouseClick(self.mask_widget.shrink_btn, QtCore.Qt.LeftButton)
self.assertTrue(np.array_equal(previous_mask, self.mask_model._mask_data))
class MaskControllerTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.app = QtGui.QApplication([])
@classmethod
def tearDownClass(cls):
cls.app.quit()
cls.app.deleteLater()
def setUp(self):
self.working_dir = {}
self.img_model = ImgModel()
self.mask_model = MaskModel()
self.mask_widget = MaskWidget()
self.mask_controller = MaskController(self.working_dir,
self.mask_widget, self.img_model,
self.mask_model)
def tearDown(self):
del self.img_model
del self.mask_model
self.mask_widget.close()
del self.mask_widget
del self.mask_controller
gc.collect()
def get_file_size(self, filename):
stat_info = os.stat(filename)
return stat_info.st_size
def test_loading_and_saving_mask_files(self):
self.mask_controller.load_mask_btn_click(
os.path.join(data_path, 'test.mask'))
self.mask_model.mask_below_threshold(self.img_model, 1)
filename = os.path.join(data_path, 'dummy.mask')
self.mask_controller.save_mask_btn_click(filename)
self.assertTrue(os.path.exists(filename))
os.remove(filename)
def test_loading_and_saving_with_super_sampling(self):
self.mask_model.set_supersampling(2)
self.mask_controller.load_mask_btn_click(
os.path.join(data_path, 'test.mask'))
self.assertEqual(self.mask_model.get_mask().shape[0], 4096)
self.assertEqual(self.mask_model.get_mask().shape[1], 4096)
self.assertEqual(self.mask_model.get_img().shape[0], 2048)
self.assertEqual(self.mask_model.get_img().shape[1], 2048)
self.mask_model.mask_below_threshold(self.img_model, 1)
filename = os.path.join(data_path, 'dummy.mask')
self.mask_controller.save_mask_btn_click(filename)
self.assertAlmostEqual(self.get_file_size(filename),
self.get_file_size(
os.path.join(data_path, 'test.mask')),
delta=20)
def test_grow_and_shrinking(self):
self.mask_model.mask_ellipse(100, 100, 20, 20)
previous_mask = np.copy(self.mask_model._mask_data)
QTest.mouseClick(self.mask_widget.grow_btn, QtCore.Qt.LeftButton)
self.assertFalse(
np.array_equal(previous_mask, self.mask_model._mask_data))
QTest.mouseClick(self.mask_widget.shrink_btn, QtCore.Qt.LeftButton)
self.assertTrue(
np.array_equal(previous_mask, self.mask_model._mask_data))
