def test_power3_step(self):
vals = np.linspace(5.0, 12.0, num=8)
self.assertEqual(
power3_step(vals, edges=[0.0, 1.0]).tolist(), [1.0] * 8)
self.assertEqual(
power3_step(vals, edges=[7.0, 11.0]).tolist(),
[0.0, 0.0, 0.0, 0.15625, 0.5, 0.84375, 1.0, 1.0])
def test_power3_step(self):
vals = np.linspace(5.0, 12.0, num=8)
self.assertEqual(power3_step(vals, edges=[0.0, 1.0]).tolist(), [1.0]*8)
self.assertEqual(power3_step(vals, edges=[7.0, 11.0]).tolist(),
[0.0, 0.0, 0.0, 0.15625, 0.5, 0.84375, 1.0, 1.0])