def test_emits_warning_for_altitude_outside_range(recwarn):
h = -1.0 # m
atm(h)
warning = recwarn.pop(RuntimeWarning)
assert issubclass(warning.category, RuntimeWarning)
def test_results_under_11km():
h = np.array([0.0,
50.0,
550.0,
6500.0,
10000.0,
11000.0
]) # m
expected_T = np.array([288.150,
287.825,
284.575,
245.900,
223.150,
216.650
]) # K
expected_p = np.array([101325.0,
100720.0,
94890.0,
44034.0,
26436.0,
22632.0
]) # Pa
expected_rho = np.array([1.2250,
1.2191,
1.1616,
0.62384,
0.41271,
0.36392
]) # kg / m3
T, p, rho = atm(h)
assert_array_almost_equal(T, expected_T, decimal=3)
assert_array_almost_equal(p, expected_p, decimal=-1)
assert_array_almost_equal(rho, expected_rho, decimal=4)
def test_results_under_32km():
h = np.array([22100,
24000,
28800,
32000
]) # m
expected_T = np.array([218.750,
220.650,
225.450,
228.650
]) # K
expected_p = np.array([3937.7,
2930.4,
1404.8,
868.01
]) # Pa
expected_rho = np.array([0.062711,
0.046267,
0.021708,
0.013225
]) # kg / m3
T, p, rho = atm(h)
assert_array_almost_equal(T, expected_T, decimal=3)
assert_array_almost_equal(p, expected_p, decimal=1)
assert_array_almost_equal(rho, expected_rho, decimal=5)
def test_results_under_20km():
h = np.array([12000,
14200,
17500,
20000
]) # m
expected_T = np.array([216.650,
216.650,
216.650,
216.650,
]) # K
expected_p = np.array([19330.0,
13663.0,
8120.5,
5474.8
]) # Pa
expected_rho = np.array([0.31083,
0.21971,
0.13058,
0.088035
]) # kg / m3
T, p, rho = atm(h)
assert_array_almost_equal(T, expected_T, decimal=3)
assert_array_almost_equal(p, expected_p, decimal=0)
assert_array_almost_equal(rho, expected_rho, decimal=5)
def test_values_outside_range_are_nan():
h = np.array([-1.0, 0.0]) # m
T, p, rho = atm(h)
assert_equal(T[0], np.nan)
assert_equal(p[0], np.nan)
assert_equal(rho[0], np.nan)
def test_scalar_input_returns_scalar_output():
h = 0.0 # m
T, p, rho = atm(h)
# Reads: "Assert if T is a float"
assert isinstance(T, float)
assert isinstance(p, float)
assert isinstance(rho, float)
def test_array_input_returns_array_output():
num = 5
h = np.zeros(5) # m
T, p, rho = atm(h)
# Reads: "Assert if the length of T equals num"
# Notice that T has to be a sequence in the first place or len(T)
# will raise TypeError
assert_equal(len(T), num)
assert_equal(len(p), num)
assert_equal(len(rho), num)
def test_sea_level():
h = 0.0 # m
expected_T = 288.15 # K
expected_p = 101325.0 # Pa
expected_rho = 1.2250 # kg / m3
T, p, rho = atm(h)
# Reads: "Assert if T equals expected_T"
assert_equal(T, expected_T)
assert_equal(p, expected_p)
assert_almost_equal(rho, expected_rho, decimal=4)
