cuvarbase.tests package

Submodules

cuvarbase.tests.test_bls module

class cuvarbase.tests.test_bls.TestBLS

Bases: future.types.newobject.newobject

atol = 1e-05

plot = False

rand = <mtrand.RandomState object>

rtol = 0.001

seed = 100

test_custom(freq, q_index, phi_index, freq_batch_size, nstreams)

test_fast_eebls(freq, q, phi0, freq_batch_size, dlogq, dphi, **kwargs)

test_standard(freq, q_index, phi_index, nstreams, freq_batch_size)

test_transit(freq, use_fast, freq_batch_size, nstreams, phi0, dlogq)

test_transit_parameter_consistency(freq, phi0, dlogq, nstreams, freq_batch_size)

cuvarbase.tests.test_bls.data(seed=100, sigma=0.1, ybar=12.0, snr=10, ndata=200, freq=10.0, q=0.01, phi0=None, baseline=1.0)

cuvarbase.tests.test_bls.get_total_nbins(nbins0, nbinsf, dlogq)

cuvarbase.tests.test_bls.manual_binning(t, y, dy, freqs, nbins0, nbinsf, dlogq, phi_min, phi_max, noverlap)

for possible tests of the binning procedure. this method has not been tested!

cuvarbase.tests.test_bls.mod1(x)

cuvarbase.tests.test_bls.plot_bls_sol(t, y, dy, freq, q, phi0)

cuvarbase.tests.test_bls.transit_model(phi0, q, delta, q1=0.0)

cuvarbase.tests.test_ce module

class cuvarbase.tests.test_ce.TestCE

Bases: future.types.newobject.newobject

plot = False

test_batched_run(ndatas, batch_size, use_double, mag_bins, phase_bins, mag_overlap, phase_overlap, use_fast, shmem_lc, weighted, freq_batch_size)

test_batched_run_const_nfreq(ndatas, batch_size, use_double, mag_bins, phase_bins, mag_overlap, phase_overlap, use_fast, weighted, shmem_lc, freq_batch_size)

test_fast(freq, use_double, mag_bins, phase_bins, mag_overlap, phase_overlap, t0, balanced_magbins, weighted, shmem_lc, freq_batch_size, force_nblocks, ndata)

test_inject_and_recover(freq, use_double, mag_bins, phase_bins, mag_overlap, phase_overlap, use_fast, t0, balanced_magbins, weighted, shmem_lc, freq_batch_size)

test_large_run(make_plot=False, **kwargs)

test_multiple_datasets(ndatas, **kwargs)

test_time_shift_invariance(freq, use_double, mag_bins, phase_bins, mag_overlap, phase_overlap, use_fast, balanced_magbins, weighted, shmem_lc, freq_batch_size)

cuvarbase.tests.test_ce.assert_similar(pdg0, pdg, top=5)

cuvarbase.tests.test_ce.data(sigma=0.1, ndata=500, freq=3.0, snr=1000, t0=0.0)

cuvarbase.tests.test_lombscargle module

class cuvarbase.tests.test_lombscargle.TestLombScargle

Bases: future.types.newobject.newobject

test_against_astropy_double()

test_against_astropy_single()

test_batched_run(ndatas=5, batch_size=5, sigma=5, samples_per_peak=3, nyquist_factor=3, **kwargs)

test_batched_run_const_nfreq(make_plot=False, ndatas=27, batch_size=5, sigma=5, samples_per_peak=3, nyquist_factor=3, **kwargs)

test_ls_kernel()

test_ls_kernel_direct_sums()

test_ls_kernel_direct_sums_against_python()

test_ls_kernel_direct_sums_is_consistent()

test_multiple_datasets(ndatas=5)

cuvarbase.tests.test_lombscargle.assert_similar(pdg0, pdg, top=5)

cuvarbase.tests.test_lombscargle.data(seed=100, sigma=0.1, ndata=100, freq=3.0)

cuvarbase.tests.test_nfft module

class cuvarbase.tests.test_nfft.TestNFFT

Bases: future.types.newobject.newobject

nfft_against_direct_sums(samples_per_peak=1, f0=None, scaled=True)

test_fast_gridding_against_scalar_version()

test_fast_gridding_with_jvdp_nfft()

test_ffts()

test_nfft_adjoint_async(f0=0.0, ndata=10, batch_size=3, use_double=False)

test_nfft_against_existing_impl_scaled_centered_spp1()

test_nfft_against_existing_impl_scaled_centered_spp5()

test_nfft_against_existing_impl_scaled_uncentered_spp1()

test_nfft_against_existing_impl_unscaled_centered_spp1()

test_nfft_against_existing_impl_unscaled_uncentered_spp5()

test_slow_gridding_against_jvdp_nfft()

test_slow_gridding_against_scalar_fast_gridding()

cuvarbase.tests.test_nfft.data(seed=100, sigma=0.1, ndata=100, samples_per_peak=1)

cuvarbase.tests.test_nfft.direct_sums(t, y, freqs)

cuvarbase.tests.test_nfft.get_b(sigma, m)

cuvarbase.tests.test_nfft.get_cpu_grid(t, y, nf, sigma=5, m=8)

cuvarbase.tests.test_nfft.gpu_grid_scalar(t, y, sigma, m, N)

cuvarbase.tests.test_nfft.precomp_psi(t, b, n, m)

cuvarbase.tests.test_nfft.scale_time(t, samples_per_peak)

cuvarbase.tests.test_nfft.simple_gpu_nfft(t, y, nf, sigma=5, use_double=False, m=8, samples_per_peak=1, **kwargs)

cuvarbase.tests.test_pdm module

cuvarbase.tests.test_pdm.data(seed=100, sigma=0.1, ndata=250)

cuvarbase.tests.test_pdm.test_cuda_pdm()

Module contents