def test_firfilter_1d_segments(self):
my_firfilter = firfilter(self.b, self.x2, x_segmented=True)
assert_array_almost_equal(my_firfilter(self.b, 0), self.x2fb[0])
assert_array_almost_equal(my_firfilter(self.b, 1), self.x2fb[1])
def test_firfilter_2d_segments(self):
my_firfilter = firfilter(self.b, self.x3, x_segmented=True)
assert_array_almost_equal(my_firfilter(self.b, 0), self.x2fb)
assert_array_almost_equal(my_firfilter(self.b, 1),
self.x2fb[::-1, :])
def test_firfilter_2d_input(self):
my_firfilter = firfilter(self.b, self.x2)
assert_array_almost_equal(my_firfilter(self.b, self.x2), self.x2fb)
def test_firfilter_2d_input_b_constant(self):
my_firfilter = firfilter(self.b, self.x2, constant_filter=True)
assert_array_almost_equal(my_firfilter(self.x2), self.x2fb)
def test_firfilter_3d_input_alternative_axis(self):
my_firfilter = firfilter(self.b, self.x3, axis=1)
assert_array_almost_equal(my_firfilter(self.b, self.x3), self.x3fb)
def test_firfilter_1d_input_x_constant(self):
my_firfilter = firfilter(self.b, self.x, constant_signal=True)
assert_array_almost_equal(my_firfilter(self.b), self.xfb)
def test_firfilter_2d_input_alternative_axis(self):
my_firfilter = firfilter(self.b, self.x2.T, axis=0)
assert_array_almost_equal(my_firfilter(self.b, self.x2.T), self.x2fb.T)
def test_firfilter_2d_input_default_axis(self):
my_firfilter = firfilter(self.b, self.x2)
assert_array_almost_equal(my_firfilter(self.b, self.x2), self.x2fb)
assert_array_almost_equal(my_firfilter(self.b, self.x2), self.x2fb)
def test_firfilter_1d_input_signal_length_error(self):
my_firfilter = firfilter(self.b, len(self.x) + 2)
with pytest.raises(ValueError):
assert_array_almost_equal(my_firfilter(self.b, self.x), self.xfb)
def test_firfilter_1d_input_signal_length(self):
my_firfilter = firfilter(self.b, len(self.x))
assert_array_almost_equal(my_firfilter(self.b, self.x), self.xfb)
def test_firfilter_1d_input(self):
my_firfilter = firfilter(self.b, self.x)
assert_array_almost_equal(my_firfilter(self.b, self.x), self.xfb)
