def setUp(self): self.function = TrapezoidalFunction( Point(0, 0), Point(6, 0), Point(2, 0.8), Point(4, 0.8) )
class TrapezoidalTests(unittest.TestCase): def setUp(self): self.function = TrapezoidalFunction( Point(0, 0), Point(6, 0), Point(2, 0.8), Point(4, 0.8) ) def test_evaluate_equal_to_A(self): self.assertEqual(self.function.evaluate(0), 0) def test_evaluate_equal_to_B(self): self.assertEqual(self.function.evaluate(6), 0) def test_evaluate_lower_than_A(self): self.assertEqual(self.function.evaluate(-1), 0) def test_evaluate_greather_than_B(self): self.assertEqual(self.function.evaluate(21), 0) def test_evaluate_between_A_C(self): self.assertEqual(self.function.evaluate(1), 0.4) def test_evaluate_between_D_B(self): self.assertEqual(self.function.evaluate(5), 0.4) def test_evaluate_equal_to_C(self): self.assertEqual(self.function.evaluate(2), 0.8) def test_evaluate_equal_to_D(self): self.assertEqual(self.function.evaluate(4), 0.8) def test_evaluate_between_C_D(self): self.assertEqual(self.function.evaluate(3), 0.8) def test_truncate_higher(self): func = self.function.truncate(1.1) self.assertEqual(func.a, self.function.a) self.assertEqual(func.b, self.function.b) self.assertEqual(func.c, self.function.c) self.assertEqual(func.d, self.function.d) def test_truncate_middle(self): func = self.function.truncate(0.4) self.assertEqual(func.a, self.function.a) self.assertEqual(func.b, self.function.b) self.assertEqual(func.c, Point(1, 0.4)) self.assertEqual(func.d, Point(5, 0.4)) def test_truncate_zero(self): func = self.function.truncate(0.4) self.assertEqual(func.a, self.function.a) self.assertEqual(func.b, self.function.b) self.assertEqual(func.c, self.function.c) self.assertEqual(func.d, self.function.d)