def test_strictly_increasing(self): data = np.linspace(1, 2., 10) self.assertEqual(zigzag.identify_initial_pivot(data, 0.1, -0.1), VALLEY)
def test_strictly_decreasing_under_threshold(self): data = np.linspace(1, 0.99, 10) self.assertEqual(zigzag.identify_initial_pivot(data, 0.1, -0.1), PEAK)
def test_decreasing_under_threshold_kinked(self): data = np.array([1.0, 1.01, 0.99]) self.assertEqual(zigzag.identify_initial_pivot(data, 0.1, -0.1), PEAK)
def test_increasing_under_threshold_kinked(self): data = np.array([1.0, 0.99, 1.02]) self.assertEqual(zigzag.identify_initial_pivot(data, 0.1, -0.1), VALLEY)
def test_decreasing_kinked(self): data = np.array([1.0, 1.01, 0.9]) self.assertEqual(zigzag.identify_initial_pivot(data, 0.1, -0.1), VALLEY)
def test_increasing_kinked(self): data = np.array([1.0, 0.99, 1.1]) self.assertEqual(zigzag.identify_initial_pivot(data, 0.1, -0.1), PEAK)