def test_time_to_x_nonlinear():
image = np.zeros((200, 3600))
spectrogram = Spectrogram(
image, np.linspace(0, image.shape[1] - 1, image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 10, 10), datetime(2010, 10, 10, 1))
ret = spectrogram.time_to_x(datetime(2010, 10, 10, 0, 0, 59))
assert isinstance(ret, int)
assert ret == 59
def test_time_to_x_nonlinear():
image = np.zeros((200, 3600))
spectrogram = Spectrogram(
image, np.linspace(0, image.shape[1] - 1, image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 10, 10), datetime(2010, 10, 10, 1)
)
ret = spectrogram.time_to_x(datetime(2010, 10, 10, 0, 0, 59))
assert isinstance(ret, int)
assert ret == 59
def test_rescale_error2():
image = np.random.rand(200, 3600) * 43
spec = Spectrogram(image, np.linspace(0, image.shape[1] - 1,
image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 1, 1, 0, 15),
datetime(2010, 1, 1, 0, 30))
with pytest.raises(ValueError) as excinfo:
spec.rescale(1, 1)
assert excinfo.value.message == "Maximum and minimum must be different."
def test_rescale_error2():
image = np.random.rand(200, 3600) * 43
spec = Spectrogram(image,
np.linspace(0, image.shape[1] - 1, image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 1, 1, 0, 15),
datetime(2010, 1, 1, 0, 30)
)
with pytest.raises(ValueError) as excinfo:
spec.rescale(1, 1)
assert excinfo.value.message == "Maximum and minimum must be different."
def test_rescale_error():
image = np.zeros((200, 3600))
spec = Spectrogram(image, np.linspace(0, image.shape[1] - 1,
image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 1, 1, 0, 15),
datetime(2010, 1, 1, 0, 30))
with pytest.raises(ValueError) as excinfo:
spec.rescale(0, 1)
assert (excinfo.value.message ==
"Spectrogram needs to contain distinct values.")
def test_rescale():
image = np.random.rand(200, 3600) * 43
spec = Spectrogram(image, np.linspace(0, image.shape[1] - 1,
image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 1, 1, 0, 15),
datetime(2010, 1, 1, 0, 30))
nspec = spec.rescale()
assert dict_eq(spec._get_params(), nspec._get_params())
assert_array_almost_equal(nspec.max(), 1)
assert nspec.min() == 0
def test_rescale_error():
image = np.zeros((200, 3600))
spec = Spectrogram(image,
np.linspace(0, image.shape[1] - 1, image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 1, 1, 0, 15),
datetime(2010, 1, 1, 0, 30)
)
with pytest.raises(ValueError) as excinfo:
spec.rescale(0, 1)
assert (
excinfo.value.message ==
"Spectrogram needs to contain distinct values.")
def test_rescale():
image = np.random.rand(200, 3600) * 43
spec = Spectrogram(image,
np.linspace(0, image.shape[1] - 1, image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 1, 1, 0, 15),
datetime(2010, 1, 1, 0, 30)
)
nspec = spec.rescale()
assert dict_eq(spec._get_params(), nspec._get_params())
assert_array_almost_equal(nspec.data.max(), 1)
assert nspec.data.min() == 0
def test_auto_t_init():
image = np.random.rand(200, 3600)
assert Spectrogram(image, np.linspace(0, image.shape[1] - 1,
image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 1, 1, 0, 15),
datetime(2010, 1, 1, 0, 30)).t_init == 900
def mk_spec(image):
return Spectrogram(
image, np.linspace(0, image.shape[1] - 1, image.shape[1]),
np.linspace(0, image.shape[0] - 1, image.shape[0]),
datetime(2010, 10, 10), datetime(2010, 10, 10, 1), 0
)
