def test_value(): w1 = Waveform(array([1,2,3]),array([3,5,6])) assert_almost_equal(w1.value(1.5), 4.0) ## 2-d waveform w2 = Waveform([[1,2],[2,3,4]], array([[3,5,6], [4,6,7]])) assert_waveform_almost_equal(w2.value(2.5), Waveform([[1, 2]], array([ 4., 5.]))) assert_waveform_almost_equal(w2.value(1.5, axis=0), Waveform([[2, 3, 4]], array([ 3.5, 5.5, 6.5]))) ## x is a waveform w2 = Waveform([[1,2],[2,3,4]], array([[3,5,6], [4,6,7]])) assert_waveform_almost_equal(w2.value(Waveform(array([1, 2]), array([2.5, 3.5]))), Waveform(array([1, 2]),array([ 4., 6.5])))