def test_outdate_valid_guzi(self): user = User("id", None) user.guzi_wallet.append("2000-01-01-id-guzi0000") user.guzi_wallet.append("2000-01-01-id-guzi0001") user.guzi_wallet.append("2000-01-01-id-guzi0002") user.outdate(["2000-01-01-id-guzi0000"]) self.assertEqual(len(user.guzi_wallet), 2) self.assertEqual(len(user.total_accumulated), 1)
def test_outdate_raise_error_for_unexisting_guzi(self): user = User("id", None) with self.assertRaises(ValueError): user.outdate([1234, 4321])