-
Notifications
You must be signed in to change notification settings - Fork 5
/
test_convert_to_unit.py
34 lines (32 loc) · 1.57 KB
/
test_convert_to_unit.py
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
#####################################################################################
# Regression test for convert_to_unit in pyspeckit.spectrum.units.SpectroscopicAxis #
# Uses doppler_radio, _optical and _relativistic equivalencies #
#####################################################################################
from pyspeckit.spectrum.units import SpectroscopicAxis
import astropy.units as u
import random
import numpy as np
import inspect
import pytest
def test_convert_to_unit(run_with_assert=False):
for from_unit, to_unit, _, __ in u.doppler_optical(1*u.Hz):
sp = SpectroscopicAxis(np.linspace(-100,100,100),
unit=from_unit, refX=23.2*u.Hz,
velocity_convention='optical')
sp.convert_to_unit(to_unit)
if(run_with_assert):
assert sp.unit == to_unit
for from_unit, to_unit, _, __ in u.doppler_radio(1*u.eV):
sp = SpectroscopicAxis(np.linspace(-100,100,100),
unit=from_unit, refX=23.2*u.Hz,
velocity_convention='optical')
sp.convert_to_unit(to_unit)
if(run_with_assert):
assert sp.unit == to_unit
for from_unit, to_unit, _, __ in u.doppler_relativistic(1*u.Angstrom):
sp = SpectroscopicAxis(np.linspace(-100,100,100),
unit=from_unit, refX=23.2*u.Hz,
velocity_convention='relativistic')
sp.convert_to_unit(to_unit)
if(run_with_assert):
assert sp.unit == to_unit