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'))