def test_asymptotic_relation(): """Test for method to compute frequencies from the asymptotic relation""" # setup norders = cs.pars['norders'] nsamples = cs.pars['nsamples'] mod = asymp_spec_model(cs.st.f, norders) dnu, numax, eps, alpha = 3050.0, 135.0, 1.45, 10**-2.5 # roughly solar func = mod._asymptotic_relation inp = [dnu, numax, eps, alpha, norders] inp_arr = [ np.float64(numax).repeat(nsamples), np.float64(dnu).repeat(nsamples), np.float64(eps).repeat(nsamples), np.float64(alpha).repeat(nsamples), norders ] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, np.ndarray) pbt.right_shape(func, inp, (norders, )) pbt.right_shape(func, inp_arr, (norders, nsamples)) # check that function returns the same as during testing given the above # inputs assert_allclose(func(*inp), [2896.028668, 3030.75434], atol=0.001)
def test_get_nmax(): """Test for method to get nmax""" # setup nsamples = cs.pars['nsamples'] norders = cs.pars['norders'] mod = asymp_spec_model(cs.st.f, norders) dnu, numax, eps = 10**cs.pars['asypars'][0], 10**cs.pars['asypars'][ 1], cs.pars['asypars'][2] func = mod._get_nmax inp = [dnu, numax, eps] inp_arr = [ np.float64(dnu).repeat(nsamples), np.float64(numax).repeat(nsamples), np.float64(eps).repeat(nsamples) ] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, float) pbt.right_type(func, inp_arr, np.ndarray) pbt.right_shape(func, inp, ()) pbt.right_shape(func, inp_arr, (nsamples, )) # check that function returns the same as during testing given the above # inputs assert (func(*inp) == 0.0)
def test_outpath(): """Tests for the function that sets the output filename Notes ----- If adding more star inits, remember to os.rmdir(st.path) to clean up after the test. """ # setup pg = lk.periodogram.Periodogram(np.array([1,1])*units.microhertz, units.Quantity(np.array([1,1]), None)) st = star('thisisatest', pg, (220.0, 3.0), (16.97, 0.05), (4750, 250), (1.34, 0.1)) func = st._outpath inp = ['test.png'] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, str) # Some input tests assert(os.path.isdir(os.path.dirname(func(*inp)))) # cleanup os.rmdir(st.path)
def test_get_priorpath(): """Tests the function for getting the default prior_data filepath""" # setup func = get_priorpath # simple tests pbt.does_it_run(func, None) pbt.does_it_return(func, None) pbt.right_type(func, None, str) # check some inputs assert ('prior_data.csv' in func()) assert ('data' in func())
def test_get_asy_start(): """Test for getting starting location of asy fit""" # setup func = cs.st.asy_fit._get_asy_start # simple tests pbt.does_it_run(func, None) pbt.does_it_return(func, None) pbt.right_type(func, None, list) pbt.right_shape(func, None, (10, )) # check function returns expected values assert_allclose(func(), [10, 10, 1, 10, 10, 1, 1, 1, 10, 1])
def test_get_freq_range(): """Test for getting frequency range of modes""" # setup func = cs.st.asy_fit._get_freq_range # simple tests pbt.does_it_run(func, None) pbt.does_it_return(func, None) pbt.right_type(func, None, tuple) pbt.right_shape(func, None, (2, )) # check function returns expected values assert_allclose(func(), [-12.5, 22.5])
def test_normal(): """Test for the log of a normal distribution""" # setup func = normal inp = [0, 0, 1] inp_arr = [np.linspace(-10, 10, 100), 0, 1] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, float) pbt.right_shape(func, inp_arr, (len(inp_arr[0]), )) # check height at mean is correct assert (10**func(*inp) == 1)
def test_select_prior_data(): """Tests for kde.select_prior_data""" # Setup KDEsize = 87 func = simp_kde.select_prior_data # Simplest run test. Given no numax it should return a sample that equals # the whole prior_data.csv pbt.does_it_run(func, None) assert(len(simp_kde.prior_data) == len(pdata)) # With a given numax and KDEsize, should return KDEsize elements in the # prior_data attribute func(to_log10(30, 10), KDEsize) assert(len(simp_kde.prior_data) == KDEsize)
def test_asymp_spec_model_call(): """Test call method for asymptotic relation spectrum model class""" # setup norders = cs.pars['norders'] mod = asymp_spec_model([0.5, 1], norders) func = mod.model inp = cs.pars['asypars'] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, np.ndarray) pbt.right_shape(func, inp, np.shape(cs.st.f)) # check that the function doesn't change the output assert_allclose(mod(inp), mod.model(*inp))
def test_to_log10(): """Test for the log of a normal distribution""" # setup func = to_log10 inp = [10, 1] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, list) # check some inputs assert (func(*inp)[0] == 1) assert_almost_equal(func(*inp)[1], 0.043429, 5) inp = [1e5, 1e4] assert (func(*inp)[0] == 5)
def test_model(): """Test for method to compute the total asymptotic relation model""" # setup norders = cs.pars['norders'] mod = asymp_spec_model([0.5, 1], norders) func = mod.model inp = cs.pars['asypars'] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, np.ndarray) pbt.right_shape(func, inp, np.shape(cs.st.f)) # check that function returns the same as during testing given the above # inputs assert_allclose(func(*inp), [7.93069307, 7.73076923], atol=0.001)
def test_pair(): """Test for method to compute the mode pairs""" # setup norders = cs.pars['norders'] mod = asymp_spec_model([0.5, 1], norders) h, freq, w, d02 = 1, cs.st.f[0], 1, 0.5 func = mod._pair inp = [h, freq, w, d02] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, np.ndarray) pbt.right_shape(func, inp, np.shape(cs.st.f)) # check that function returns the same as during testing given the above # inputs assert_array_equal(func(*inp), [1.2, 1.35])
def test_lor(): """Test for method to compute the lorentzian profiles""" # setup norders = cs.pars['norders'] mod = asymp_spec_model(cs.st.f, norders) h, freq, w = 1, cs.st.f[0], 1 func = mod._lor inp = [freq, h, w] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, np.ndarray) pbt.right_shape(func, inp, np.shape(cs.st.f)) # check that function returns the same as during testing given the above # inputs assert_array_equal(func(*inp), [1, 1])
def test_P_envelope(): """Test for method to get height of pmode envelope""" # setup norders = cs.pars['norders'] mod = asymp_spec_model(cs.st.f, norders) nus = cs.pars['freqs'] envheight, numax, envwidth = 10**1.5, nus[0], 10**2.2 func = mod._P_envelope inp = [nus, envheight, numax, envwidth] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, np.ndarray) pbt.right_shape(func, inp, np.shape(nus)) # check that function returns the same as during testing given the above # inputs assert_allclose(func(*inp), [31.6227766, 31.5718312], atol=0.001)
def test_get_percentiles(): """Tests the function for getting the percentiless of a distribution""" # setup func = get_percentiles inp = [np.random.normal(0, 1, size=10), 3] #print(func(*inp)[]) # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, np.ndarray) pbt.right_shape(func, inp, (2 * inp[1] + 1, )) # check some different inputs inp = [np.random.normal(0, 1, size=30000), 5] pbt.right_shape(func, inp, (2 * inp[1] + 1, )) inp = [[0, 0, 0, 1, 1], 1] assert_array_equal(func(*inp), [0., 0., 1.])
def test_run_kde(): """Tests that a KDE can be created and sampled for given prior data Notes ----- This test is incomplete """ # setup pg = lk.periodogram.Periodogram( np.array([1, 1]) * units.microhertz, units.Quantity(np.array([1, 1]), None)) st = star('thisisatest', pg, (220.0, 3.0), (16.97, 0.05), (4750, 250), (1.34, 0.1)) func = st.run_kde # simple tests pbt.does_it_run(func, None) # cleanup os.rmdir(st.path)
def test_get_enns(): """Test for method to get the radial orders in the asymptotic relation""" # setup norders = cs.pars['norders'] nmax = cs.pars['nmax'] nsamples = cs.pars['nsamples'] mod = asymp_spec_model(cs.st.f, norders) func = mod._get_enns inp = [nmax, norders] inp_arr = [np.array([nmax]).repeat(nsamples), norders] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, np.ndarray) pbt.right_shape(func, inp, (norders, )) pbt.right_shape(func, inp_arr, (nsamples, norders)) # check that function returns the same as during testing given the above # inputs assert_array_equal(func(*inp), [0, 1])
def test_asymp_spec_model_call(): """Test call method for asymptotic relation spectrum model class""" # setup norders = cs.pars['norders'] mod = asymp_spec_model([0.5, 1], norders) func = mod.model inp = cs.pars['asypars'] # simple tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, np.ndarray) pbt.right_shape(func, inp, np.shape(cs.st.f)) # check that the function doesn't change the output assert_allclose(mod(inp), mod.model(*inp)) # The test functions below require longer runs and are not suitable for GitHub # workflows. the mark.slow decorator doesn't seem to work with GitHub workflows. # #def test_asymptotic_fit_call():
def test_kde_predict(): """Tests for the kde.kde_predict method""" # Test raise statement with pytest.raises(ValueError): kde().kde_predict(8) # Setup err = 0.01 simp_kde.samples = np.vstack([np.random.normal(m, err*abs(m), 50) for m in solar_p]).T enns = range(15,25) func = simp_kde.kde_predict inp = [enns] # basic run/return/type/shape tests pbt.does_it_run(func, inp) pbt.does_it_return(func, inp) pbt.right_type(func, inp, tuple) pbt.right_shape(func, inp, (2, 10)) # Test that errors propogate more or less correctly (probably not the most # robust test) out = simp_kde.kde_predict(enns) assert_almost_equal(out[1]/out[0], np.ones_like(out[1])*err, decimal = 1)