def test_lookup(self): d = Deck() card = d.lookup("R5")[0] self.assertEqual(card.color, "R") self.assertEqual(card.value, "5") card = d.lookup("NW")[0] self.assertEqual(card.color, "N") self.assertEqual(card.value, "W") self.assertRaises(InvalidCard, d.lookup, "AA") card = d.lookup("NW")[0] card.status = 2 self.assertEqual(len(d.lookup("NW")), 3)