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
50
51
52
53
54
55
56
57
58
59
|
import pytest
from astropy import units as u
from astropy.tests.helper import assert_quantity_allclose
from astropy.time import Time
from poliastro.util import time_range
def test_time_range_spacing_periods():
start_time = "2017-10-12 00:00:00"
end_time = "2017-10-12 00:04:00"
spacing = 1 * u.minute
periods = 5
expected_scale = "utc"
expected_duration = 4 * u.min
result_1 = time_range(start_time, spacing=spacing, periods=periods)
result_2 = time_range(start_time, end=end_time, periods=periods)
result_3 = time_range(
Time(start_time), end=Time(end_time), periods=periods
)
assert len(result_1) == len(result_2) == len(result_3) == periods
assert result_1.scale == result_2.scale == result_3.scale == expected_scale
assert_quantity_allclose(
(result_1[-1] - result_1[0]).to(u.s), expected_duration
)
assert_quantity_allclose(
(result_2[-1] - result_2[0]).to(u.s), expected_duration
)
assert_quantity_allclose(
(result_3[-1] - result_3[0]).to(u.s), expected_duration
)
def test_time_range_requires_keyword_arguments():
with pytest.raises(TypeError) as excinfo:
time_range(0, 0) # type: ignore
assert (
"TypeError: time_range() takes 1 positional argument but"
in excinfo.exconly()
)
def test_time_range_raises_error_wrong_arguments():
exception_message = (
"ValueError: Either 'end' or 'spacing' must be specified"
)
with pytest.raises(ValueError) as excinfo_1:
time_range("2017-10-12 00:00")
with pytest.raises(ValueError) as excinfo_2:
time_range("2017-10-12 00:00", spacing=0, end=0, periods=0)
assert exception_message in excinfo_1.exconly()
assert exception_message in excinfo_2.exconly()
|
