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))