skymodel/test_skymodel.py at master · dlakaplan/skymodel · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
from __future__ import print_function
import numpy as np

from astropy.coordinates import SkyCoord
from astropy import units as u, constants as c
from skymodel import skymodel

source = SkyCoord("05h35m17.3s -05d23m28s", frame='icrs')

d=skymodel.SkyModel()
print(d.distance(source, 10)[0])
print(d.DM(source, 659*u.pc)[0])
print(d.Tsky(source))

print(d.distance('0038-2501.par', 10)[0])
print(d.DM('0038-2501.par', 449*u.pc)[0])
print(d.Tsky('0038-2501.par'))

ra=np.linspace(0,240,15).reshape((5,3))
dec=np.linspace(-30,30,15).reshape((5,3))

source=SkyCoord(ra,dec,unit='deg')
try:
    print(d.Tsky(source))
except TypeError:
    print('Error converting to Galactic coords: probably astropy/numpy bug')
try:
    print(d.DM(source, 1*u.kpc)[0])
except TypeError:
    print('Error converting to Galactic coords: probably astropy/numpy bug')
try:
    print(d.distance(source, 10)[0])
except TypeError:
    print('Error converting to Galactic coords: probably astropy/numpy bug')


d=skymodel.SkyModel(dmmodel='YMW16')
print(d.distance(source, 10)[0])
print(d.DM(source, 659*u.pc)[0])
print(d.Tsky(source))

print(d.distance('0038-2501.par', 10)[0])
print(d.DM('0038-2501.par', 449*u.pc)[0])
print(d.Tsky('0038-2501.par'))


print(skymodel.Tsky(source))
print(skymodel.DM(source, 10*u.pc)[0])
print(skymodel.distance(source, 10)[0])