class TestRat(unittest.TestCase):
def setUp(self):
self.rat = Rat('P', 1, 4)
def test_rat_create(self):
self.assertEqual(self.rat.symbol, 'P')
self.assertEqual(self.rat.row, 1)
self.assertEqual(self.rat.col, 4)
self.assertEqual(self.rat.num_sprouts_eaten, 0)
def test_set_location(self):
self.rat.set_location(5,7)
self.assertEqual(self.rat.row, 5)
self.assertEqual(self.rat.col, 7)
def test_num_sprouts_eaten(self):
self.rat.eat_sprout()
self.assertEqual(self.rat.num_sprouts_eaten, 1)
self.rat.eat_sprout()
self.assertEqual(self.rat.num_sprouts_eaten, 2)
def test_print(self):
rat = Rat('J', 4, 3)
rat.eat_sprout()
rat.eat_sprout()
self.assertEqual(rat.__str__(), 'J at (4, 3) ate 2 sprouts.')
def test_Rat_set_location():
rat = Rat("P", 1, 4)
rat.set_location(3, 2)
assert rat.row == 3
assert rat.col == 2
rat.set_location(1, 3)
assert rat.row == 1
assert rat.col == 3
