def setUp(self): self.ui_core = UICore(0, 0, 2, 1, 1000)
class TestUICore(unittest.TestCase): def setUp(self): self.ui_core = UICore(0, 0, 2, 1, 1000) def test_select_object_type(self): self.ui_core.add_object("id1", "type1", 0.5, 0.5) self.ui_core.add_object("id2", "type2", 0.5, 0.5) self.ui_core.add_object("id3", "type2", 0.5, 0.5) self.ui_core.select_object_type("type1") self.assertEquals(self.ui_core.get_object("id1").selected, True, "test_select_object_type") self.assertEquals(self.ui_core.get_object("id2").selected, False, "test_select_object_type") self.assertEquals(self.ui_core.get_object("id3").selected, False, "test_select_object_type") self.ui_core.select_object_type("type2") self.assertEquals(self.ui_core.get_object("id1").selected, False, "test_select_object_type") self.assertEquals(self.ui_core.get_object("id2").selected, True, "test_select_object_type") self.assertEquals(self.ui_core.get_object("id3").selected, True, "test_select_object_type") self.ui_core.select_object_type("type1", unselect_others=False) self.assertEquals(self.ui_core.get_object("id1").selected, True, "test_select_object_type") self.assertEquals(self.ui_core.get_object("id2").selected, True, "test_select_object_type") self.assertEquals(self.ui_core.get_object("id3").selected, True, "test_select_object_type") def test_get_object(self): self.ui_core.add_object("id1", "type1", 0.5, 0.5) self.ui_core.add_object("id2", "type1", 0.5, 0.5) self.assertIsNotNone(self.ui_core.get_object("id1"), "test_get_object") self.assertEquals(self.ui_core.get_object("id1").object_id, "id1", "test_get_object") self.assertIsNone(self.ui_core.get_object("non_existent_id"), "test_get_object") def test_remove_object(self): self.ui_core.add_object("id1", "type1", 0.5, 0.5) self.ui_core.add_object("id2", "type1", 0.5, 0.5) self.assertEquals(len(list(self.ui_core.get_scene_items_by_type(ObjectItem))), 2, "test_remove_object") self.assertEquals(self.ui_core.remove_object("id1"), True, "test_remove_object") self.assertEquals(len(list(self.ui_core.get_scene_items_by_type(ObjectItem))), 1, "test_remove_object") self.assertEquals(self.ui_core.remove_object("id2"), True, "test_remove_object") self.assertEquals(len(list(self.ui_core.get_scene_items_by_type(ObjectItem))), 0, "test_remove_object") self.assertEquals(self.ui_core.remove_object("id1"), False, "test_remove_object")