Пример #1
0
    def setUp(self):

        self.ui_core = UICore(0,  0,  2, 1, 1000)
Пример #2
0
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")