def test_set_and_get_wheel(self): print("bike set/get wheel integration test. ") test = Bike("red", "BMX") if Test_bike.test_set_wheel_amount(self): if Test_bike.test_get_wheel_amount(self): if self.assertEqual(test.get_wheel_amount(), 2): return
def test_get_wheel_amount(self): print("bike get_wheel_amount test. ") test = Bike("RED", "BMX") if self.assertIsNotNone(test.get_wheel_amount()): return