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 )