def test_apply_adjustments_add(self): sale = Sale("product", value=100, amount=2) sale.add_adjustment("add", 40) expected_value = 140 expected_total_value = 280 sale.apply_adjustments() self.assertEqual(expected_value, sale.value) self.assertEqual(expected_total_value, sale.total_value)
def test_apply_adjustments_mul(self): sale = Sale("product", value=2, amount=2) sale.add_adjustment("multiply", 3) expected_value = 6 expected_total_value = 12 sale.apply_adjustments() self.assertEqual(expected_value, sale.value) self.assertEqual(expected_total_value, sale.total_value)
def test_apply_adjustments_sub(self): sale = Sale("product", value=10, amount=3) sale.add_adjustment("subtract", 40) expected_value = -30 expected_total_value = -90 sale.apply_adjustments() self.assertEqual(expected_value, sale.value) self.assertEqual(expected_total_value, sale.total_value)