def test_location(self): a = Ranch() a.set_location("loc1") self.assertEqual(a.get_location(), "loc1") a.set_location("loc2") self.assertEqual(a.get_location(), "loc2")