def test_profile1(j1713_profile): """ Test for data profiles 1. """ pr = DataProfile(j1713_profile,phases=None) pr.init_profiles(100) assert(hasattr(pr,'_profiles')) assert(hasattr(pr,'_max_profile'))
def test_profile3(j1713_profile): """ Test for data profiles 3. """ ph = np.linspace(0,1,2048) pr = DataProfile(j1713_profile,phases=ph) pr.init_profiles(100, Nchan=20) assert(hasattr(pr,'_profiles')) assert(hasattr(pr,'_max_profile'))
def test_initprof(simulation, j1713_profile): """ Test init_profile function. """ # Test no input simulation.init_profile() # Test function input with pytest.raises(NotImplementedError): def gprof(x, p0): return p0[0] * np.exp(-0.5 * ((x - p0[1]) / (p0[2]))**2) simulation._profiles = gprof simulation.init_profile() # Test Gaussian as input simulation._profiles = [0.5, 0.5, 1.0] simulation.init_profile() # Test data array as input simulation._profiles = j1713_profile simulation.init_profile() # Test array that's not long enough with pytest.raises(RuntimeError): simulation._profiles = [0.5, 0.5] simulation.init_profile() # Test profile class as input pr = DataProfile(j1713_profile, phases=None) print(type(pr), pr) simulation._profiles = pr simulation.init_profile()
def j1713_profile(): """ DataProfile of J1713+0747 profile. """ path = 'psrsigsim/data/J1713+0747_profile.npy' pr = DataProfile(np.load(path), phases=None) return pr
def j1713_profile(): """ Numpy array of J1713+0747 profile. """ path = 'psrsigsim/data/J1713+0747_profile.npy' pr = DataProfile(np.load(path), phases=None, Nchan=2) return pr
def test_profile2(j1713_profile): """ Test for data profiles 2. """ pr = DataProfile(j1713_profile,phases=None) pr.init_profiles(100, Nchan=20) assert(hasattr(pr,'_profiles')) assert(hasattr(pr,'_max_profile')) with pytest.raises(NotImplementedError): pr.set_Nchan(2)
def test_profile2(j1713_profile): pr = DataProfile(j1713_profile, phases=None) pr.init_profiles(100, Nchan=20) assert (hasattr(pr, '_profiles')) assert (hasattr(pr, '_max_profile'))