def test_expand_2d(): # 2d make wave format wave = [0, 100, 2, 1, 0, 0, 0, 4, 2, 2, 1000, 100, 0, 100, 0, 0, 0, 0, 0] wave = np.array(wave) new = echelle.expand_polynomial(2, wave) assert isinstance(new, np.ndarray) assert new.ndim == 2 assert new.shape[0] == 2 assert new.shape[1] == 100 cmp = np.linspace(1001, 1011, 100, endpoint=False) assert np.allclose(new[0], cmp) cmp = np.linspace(501, 506, 100, endpoint=False) assert np.allclose(new[1], cmp) wave = [ 0, 100, 2, 1, 0, 0, 0, 6, 3, 3, 1000, 100, 0, 0, 100, 0, 0, 0, 0, 0, 0, 0, 0 ] wave = np.array(wave) new = echelle.expand_polynomial(2, wave) assert isinstance(new, np.ndarray) assert new.ndim == 2 assert new.shape[0] == 2 assert new.shape[1] == 100 cmp = np.linspace(1001, 1011, 100, endpoint=False) assert np.allclose(new[0], cmp) cmp = np.linspace(501, 506, 100, endpoint=False) assert np.allclose(new[1], cmp)
def test_expand_1d(): wave = [[0.1, 1001], [0.05, 501]] wave = np.array(wave) new = echelle.expand_polynomial(100, wave) assert isinstance(new, np.ndarray) assert new.ndim == 2 assert new.shape[0] == 2 assert new.shape[1] == 100 cmp = np.linspace(1001, 1011, 100, endpoint=False) assert np.allclose(new[0], cmp) cmp = np.linspace(501, 506, 100, endpoint=False) assert np.allclose(new[1], cmp)