def test1(self): c = Cell() self.assertTrue(c.is_unknown()) self.assertFalse(c.is_object()) self.assertFalse(c.is_obstacle()) self.assertFalse(c.is_marked()) self.assertFalse(c.is_free()) c.set_obstacle() self.assertFalse(c.is_unknown()) self.assertFalse(c.is_object()) self.assertTrue(c.is_obstacle()) self.assertFalse(c.is_marked()) self.assertFalse(c.is_free()) self.assertFalse(c.is_blue()) self.assertFalse(c.is_red()) self.assertTrue(c.is_undef(), c.get_color_id()) self.assertTrue(c.get_color() == c.UNDEF) c.set_blue() self.assertFalse(c.is_unknown()) self.assertFalse(c.is_object()) self.assertTrue(c.is_obstacle()) self.assertFalse(c.is_marked()) self.assertFalse(c.is_free()) self.assertTrue(c.is_blue()) self.assertFalse(c.is_red()) self.assertFalse(c.is_undef()) c.set_free() self.assertFalse(c.is_unknown()) self.assertFalse(c.is_object()) self.assertFalse(c.is_obstacle()) self.assertFalse(c.is_marked()) self.assertTrue(c.is_free()) self.assertFalse(c.is_blue()) self.assertFalse(c.is_red()) c.set_object() self.assertFalse(c.is_unknown()) self.assertTrue(c.is_object()) self.assertFalse(c.is_obstacle()) self.assertFalse(c.is_marked()) self.assertFalse(c.is_free()) self.assertTrue(c.is_blue()) self.assertFalse(c.is_red()) self.assertFalse(c.is_undef()) c.set_mark() self.assertFalse(c.is_unknown()) self.assertTrue(c.is_object()) self.assertFalse(c.is_obstacle()) self.assertTrue(c.is_marked()) self.assertFalse(c.is_free()) self.assertTrue(c.is_blue()) self.assertFalse(c.is_red())
def test2_2(self): c = Cell() c.set_obstacle() c.points[Cell.UNDEF_ID] = 95 c.points[Cell.RED_ID] = 5 self.assertFalse(c.is_red(), c.get_color_id()) self.assertTrue(c.is_undef(), c.get_color_id())
def test2_3(self): c = Cell() c.set_obstacle() c.points[Cell.UNDEF_ID] = 94 c.points[Cell.RED_ID] = 6 self.assertTrue(c.is_red(), c.get_color_id()) print c.get_color_hex() self.assertEqual(c.get_color_hex(), 'ff0000') self.assertFalse(c.is_undef(), c.get_color_id())