def test_frequency_data_stub_slice():
"""Test slicing the FrequencyData stub."""
freq = [[1, 0, -1], [2, 0, -2]]
frequencies = [0, .1, .4]
frequency_data_stub = stub_utils.frequency_data_stub(freq, frequencies)
stub_slice = frequency_data_stub[0]
npt.assert_allclose(stub_slice.freq, np.atleast_2d(freq[0]))
npt.assert_allclose(stub_slice.frequencies, np.atleast_1d(frequencies))
def test_frequency_data_stub_data():
"""Test the data contained in the FrequencyData stub."""
freq = [1, 0, -1]
frequencies = [0, .1, .4]
frequency_data_stub = stub_utils.frequency_data_stub(freq, frequencies)
npt.assert_allclose(frequency_data_stub.freq, np.atleast_2d(freq))
npt.assert_allclose(frequency_data_stub.frequencies,
np.atleast_1d(frequencies))
def frequency_data_one_point():
"""
FrequencyData stub with one data points.
Returns
-------
frequency_data
stub of pyfar FrequencyData class
"""
frequency_data = stub_utils.frequency_data_stub([2], [0])
return frequency_data
def test_frequency_data_stub_properties():
""" Test comparing properties of FrequencyData stub
with actual FrequencyData implementation.
"""
freq = [1, 0, -1]
frequencies = [0, .1, .4]
frequency_data_stub = stub_utils.frequency_data_stub(freq, frequencies)
stub_dir = dir(frequency_data_stub)
frequency_data_dir = dir(FrequencyData(freq, frequencies))
assert stub_dir.sort() == frequency_data_dir.sort()