コード例 #1
0
class BicyleModelTest(unittest.TestCase):

    def setUp(self):
        self._frame = CarbonFrame(10, 100)
        self._frontWheel = RoadWheel("Front RoadWheel", 10, 100)
        self._backWheel1 = MountainWheel("Back MountainWheel", 10, 100)
        self._backWheel2 = RoadWheel("Back RoadWheel", 10, 100)
        self._bike = BicycleModel(
            "Good Build", self._frontWheel, self._backWheel2, self._frame
        )

    def test_different_heel_type_init(self):
        self.assertRaises(
            AssertionError,
            BicycleModel,
            "Error Build",
            self._frontWheel,
            self._backWheel1,
            self._frame
        )

    def test_get_cost(self):
        self.assertEqual(30, self._bike.get_cost())

    def test_get_weight(self):
        self.assertEqual(300, self._bike.get_weight())

    def test_change_cost(self):
        self._bike.change_cost_by_percentage(0.1)
        self.assertEqual(33, self._bike.get_cost())
コード例 #2
0
ファイル: unit_test.py プロジェクト: olzhas23/thinkful-mentor
class BicyleModelTest(unittest.TestCase):
    def setUp(self):
        self._frame = CarbonFrame(10, 100)
        self._frontWheel = RoadWheel("Front RoadWheel", 10, 100)
        self._backWheel1 = MountainWheel("Back MountainWheel", 10, 100)
        self._backWheel2 = RoadWheel("Back RoadWheel", 10, 100)
        self._bike = BicycleModel("Good Build", self._frontWheel,
                                  self._backWheel2, self._frame)

    def test_different_heel_type_init(self):
        self.assertRaises(AssertionError, BicycleModel, "Error Build",
                          self._frontWheel, self._backWheel1, self._frame)

    def test_get_cost(self):
        self.assertEqual(30, self._bike.get_cost())

    def test_get_weight(self):
        self.assertEqual(300, self._bike.get_weight())

    def test_change_cost(self):
        self._bike.change_cost_by_percentage(0.1)
        self.assertEqual(33, self._bike.get_cost())