def test_parse_storage_file():
"""Test parsing of the storage file with mixed TLE input."""
file_path = process_paths(mixed_tle_file_path_1)
tle_storage = TleStorage.from_path(file_path)
truth_no_of_elements = 19
truth_pos = 3
truth_name = "APRIZESAT 1"
truth_line1 = (
"1 28372U 04025G 21088.15052337 .00000024 00000-0 18112-4 0 9991"
)
truth_line2 = (
"2 28372 98.4324 42.3607 0046090 190.1598 169.8668 14.49665428885585"
)
# check number of elements
assert len(tle_storage.tle_list) == truth_no_of_elements
# check specific element at `truth_pos`
assert str(tle_storage.tle_list[truth_pos]).split("\n")[0:3] == [
truth_name,
truth_line1,
truth_line2,
]
def test_tle_param_filter_error():
with pytest.raises(ValueError):
"""Tests filtering for a TleFilterParams.TLE error."""
file_path = process_paths(mixed_tle_file_path_1)
tle_storage = TleStorage.from_path(file_path)
tle_storage.filter_by_value(TleFilterParams.TLE, 46495)
def test_tle_timeseries_unordered():
"""Test parsing of the TLE Timeseries with inverted time input."""
file_path = process_paths(time_series_tle_file_path_1)
tle_storage = TleStorage.from_path(file_path).to_tle_timeseries(37791)
assert tle_storage.tle_list[0].epoch.isot == "2021-03-30T04:20:36.404"
assert tle_storage.tle_list[-1].epoch.isot == "2021-04-01T20:16:48.785"
def test_tle_timeseries_ordered():
"""Test parsing of the TLE Timeseries with ordered time input."""
file_path = process_paths(time_series_tle_file_path_2)
tle_storage = TleStorage.from_path(file_path).to_tle_timeseries(37791)
assert tle_storage.tle_list[0].epoch.isot == "2021-03-15T02:02:02.753"
assert tle_storage.tle_list[-1].epoch.isot == "2021-03-31T21:19:42.808"
def test_tle_timeseries_time_filter():
"""Test parsing of the TLE Timeseries with time filter."""
file_path = process_paths(time_series_tle_file_path_1)
tle_storage = TleStorage.from_path(file_path).to_tle_timeseries(37791)
threshold_time = Time("2021-04-01T00:00:00.000")
tle_storage_filtered = tle_storage.filter_by_range(
TleFilterParams.EPOCH, min_value=threshold_time
)
assert isinstance(tle_storage_filtered, TleTimeSeries)
assert tle_storage_filtered.tle_list[0].epoch.isot == "2021-04-01T02:14:48.404"
assert tle_storage_filtered.tle_list[-1].epoch.isot == "2021-04-01T20:16:48.785"
def test_filter_value():
"""Tests filtering for a value equivalence."""
file_path = process_paths(mixed_tle_file_path_1)
tle_storage = TleStorage.from_path(file_path)
filtered_list_sat_nr = tle_storage.filter_by_value(
TleFilterParams.SAT_NUMBER, 46495
)
filtered_list_int_deg = tle_storage.filter_by_value(
TleFilterParams.INTL_DESIGNATOR, "18014H"
)
empty_filtered_list = tle_storage.filter_by_value(TleFilterParams.SAT_NUMBER, 56495)
assert isinstance(filtered_list_sat_nr, TleStorage)
assert tle_storage.tle_list[0].name == "APRIZESAT 2"
assert len(filtered_list_sat_nr.tle_list) == 2
assert len(filtered_list_int_deg.tle_list) == 1
assert len(empty_filtered_list.tle_list) == 0
def test_filter_func():
"""Tests filtering for a value range with a function."""
file_path = process_paths(mixed_tle_file_path_1)
tle_storage = TleStorage.from_path(file_path)
def sma_filter_1(a):
"""Semimajor axis filter min."""
return True if a > 7000 * u.km else False
def sma_filter_2(a):
"""Semimajor axis filter max."""
return True if 7000 * u.km > a else False
def sma_filter_3(a):
"""Semimajor axis filter min/max."""
return True if 7100 * u.km > a > 7000 * u.km else False
def tle_filter_1(tle):
"""Semimajor axis filter min/max."""
return True if 7100 * u.km > tle.sm_axis > 7000 * u.km else False
filtered_list_sma_1 = tle_storage.filter_by_func(
TleFilterParams.SM_AXIS, sma_filter_1
)
filtered_list_sma_2 = tle_storage.filter_by_func(
TleFilterParams.SM_AXIS, sma_filter_2
)
filtered_list_sma_3 = tle_storage.filter_by_func(
TleFilterParams.SM_AXIS, sma_filter_3
)
filtered_list_sma_4 = tle_storage.filter_by_func(TleFilterParams.TLE, tle_filter_1)
assert len(filtered_list_sma_1.tle_list) == 11
assert len(filtered_list_sma_2.tle_list) == 8
assert len(filtered_list_sma_3.tle_list) == 8
assert len(filtered_list_sma_4.tle_list) == 8
def test_filter_range():
"""Tests filtering for a value range with min/max parameters."""
file_path = process_paths(mixed_tle_file_path_1)
tle_storage = TleStorage.from_path(file_path)
filtered_list_sma_1 = tle_storage.filter_by_range(
TleFilterParams.SM_AXIS, min_value=7000 * u.km
)
filtered_list_sma_2 = tle_storage.filter_by_range(
TleFilterParams.SM_AXIS, max_value=7000 * u.km
)
filtered_list_sma_3 = tle_storage.filter_by_range(
TleFilterParams.SM_AXIS, min_value=7000 * u.km, max_value=7100 * u.km
)
filtered_list_sma_4 = tle_storage.filter_by_range(TleFilterParams.SM_AXIS)
assert len(filtered_list_sma_1.tle_list) == 11
assert len(filtered_list_sma_2.tle_list) == 8
assert len(filtered_list_sma_3.tle_list) == 8
assert len(filtered_list_sma_4.tle_list) == 0