test: update tests to use local data

doc/source/changelog/882.test.md

@@ -0,0 +1 @@
+Update tests to use local data

doc/source/conf.py

@@ -485,13 +485,15 @@ def copy_examples_to_output_dir(app: sphinx.application.Sphinx, exception: Excep
     # TODO: investigate issues when using OUTPUT_EXAMPLES instead of SOURCE_EXAMPLES
     # https://github.com/ansys/pystk/issues/415
     OUTPUT_EXAMPLES = pathlib.Path(app.outdir) / "examples"
+    OUTPUT_DATA = OUTPUT_EXAMPLES / "data"
     OUTPUT_IMAGES = OUTPUT_EXAMPLES / "img"
-    for directory in [OUTPUT_EXAMPLES, OUTPUT_IMAGES]:
+    for directory in [OUTPUT_EXAMPLES, OUTPUT_DATA, OUTPUT_IMAGES]:
         if not directory.exists():
             directory.mkdir(parents=True, exist_ok=True)
 
     SOURCE_EXAMPLES = pathlib.Path(app.srcdir) / "examples"
     EXAMPLES_DIRECTORY = SOURCE_EXAMPLES.parent.parent.parent / "examples"
+    EXAMPLES_DATA_DIRECTORY = EXAMPLES_DIRECTORY / "data"
     IMAGES_DIRECTORY = EXAMPLES_DIRECTORY / "img"
 
     # Copy the examples
@@ -508,6 +510,20 @@ def copy_examples_to_output_dir(app: sphinx.application.Sphinx, exception: Excep
         destination_file = OUTPUT_EXAMPLES / file.name
         destination_file.write_text(file.read_text(encoding="utf-8"), encoding="utf-8")
 
+    # Copy the data files
+    all_data_files = list(EXAMPLES_DATA_DIRECTORY.glob("*.tle"))
+    data_files = [file for file in all_data_files]
+    for file in status_iterator(
+        data_files,
+        f"Copying example data to doc/_build/examples/data",
+        "green",
+        len(data_files),
+        verbosity=1,
+        stringify_func=(lambda x: x.name),
+    ):
+        destination_file = OUTPUT_DATA / file.name
+        destination_file.write_text(file.read_text(encoding="utf-8"), encoding="utf-8")
+
     # Copy the static images
     images = list(IMAGES_DIRECTORY.glob("*.png"))
     for file in status_iterator(
@@ -533,15 +549,17 @@ def copy_examples_files_to_source_dir(app: sphinx.application.Sphinx):
 
     """
     SOURCE_EXAMPLES = pathlib.Path(app.srcdir) / "examples"
+    SOURCE_DATA = SOURCE_EXAMPLES / "data"
     SOURCE_IMAGES = SOURCE_EXAMPLES / "img"
-    for directory in [SOURCE_EXAMPLES, SOURCE_IMAGES]:
+    for directory in [SOURCE_EXAMPLES, SOURCE_DATA, SOURCE_IMAGES]:
         if not directory.exists():
             directory.mkdir(parents=True, exist_ok=True)
 
     EXAMPLES_DIRECTORY = SOURCE_EXAMPLES.parent.parent.parent / "examples"
+    EXAMPLES_DATA_DIRECTORY = EXAMPLES_DIRECTORY / "data"
     IMAGES_DIRECTORY = EXAMPLES_DIRECTORY / "img"
 
-    # Copy the the examples
+    # Copy the examples
     all_examples = list(EXAMPLES_DIRECTORY.glob("*.py"))
     examples = [file for file in all_examples if f"{file.name}" not in exclude_examples]
     for file in status_iterator(
@@ -555,6 +573,20 @@ def copy_examples_files_to_source_dir(app: sphinx.application.Sphinx):
         destination_file = SOURCE_EXAMPLES / file.name
         destination_file.write_text(file.read_text(encoding="utf-8"), encoding="utf-8")
 
+    # Copy the data files
+    example_data_files = list(EXAMPLES_DATA_DIRECTORY.glob("*.tle"))
+    data_files = [file for file in example_data_files]
+    for file in status_iterator(
+        data_files,
+        f"Copying example data to doc/source/examples/data/",
+        "green",
+        len(data_files),
+        verbosity=1,
+        stringify_func=(lambda file: file.name),
+    ):
+        destination_file = SOURCE_DATA / file.name
+        destination_file.write_text(file.read_text(encoding="utf-8"), encoding="utf-8")
+
     # Copy the static images
     images = list(IMAGES_DIRECTORY.glob("*.png"))
     for file in status_iterator(

examples/communication-link-calculator.py

@@ -100,9 +100,10 @@
 satellite.set_propagator_type(PropagatorType.SGP4)
 propagator = satellite.propagator
 
-# Finally, use the propagator's `common_tasks` property to add the satellite's orbit from an online source, and propagate the satellite:
+# Finally, use the propagator's `common_tasks` property to add the satellite's orbit from a local file, and propagate the satellite:
 
-propagator.common_tasks.add_segments_from_online_source("31698")
+tle_file = pathlib.Path("data") / "TerraSarX_31698.tle"
+propagator.common_tasks.add_segments_from_file("31698", str(tle_file.resolve()))
 propagator.propagate()
 
 # ## Add the camp site

examples/data/TerraSarX_31698.tle

@@ -0,0 +1,3 @@
+1 31698U 07026A   24075.25000000  .00008923  00000-0  42574-3 0 00007
+2 31698 097.4466 083.7351 0001914 083.7701 301.0238 15.19143971928808
+

examples/data/iss_5Jun2022.tle

@@ -0,0 +1,2 @@
+1 25544U 98067A   22156.00000000  .00001944  00000-0  34115-4 0 00005
+2 25544 051.6451 036.9525 0004589 195.9534 179.2523 15.49879136343299

examples/facility-to-satellite-access.py

@@ -136,18 +136,19 @@
 
 # ### Add a satellite using the SPG4 propagator
 
-# **Note:** this portion requires internet access.
-
-# First, create the satellite using online data for the International Space Station (SSN number $25544$):
+# First, create the satellite using local data for the International Space Station (SSN number $25544$):
 
 # +
+import pathlib
+
 from ansys.stk.core.stkobjects import PropagatorType
 
 
 satellite = root.current_scenario.children.new(STKObjectType.SATELLITE, "MySatellite")
 satellite.set_propagator_type(PropagatorType.SGP4)
 propagator = satellite.propagator
-propagator.common_tasks.add_segments_from_online_source("25544")
+tle_file = pathlib.Path("data") / "iss_5Jun2022.tle"
+propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
 propagator.propagate()
 # -
 

tests/doc_snippets_tests/data/KSU-CUBESAT_47954_1Jun2020.tle

@@ -0,0 +1,2 @@
+1 47954U 21022Y   20153.75000000  .00000000  00000-0  00000-0 0 00002
+2 47954 097.5686 057.1838 0018941 165.4562 124.5594 15.05622178000003

tests/doc_snippets_tests/data_analysis/data_analysis_snippets.py

@@ -23,6 +23,8 @@
 import os
 import sys
 
+from pathlib import Path
+
 
 # Add path to the parent directory to use some common utilities
 sys.path.insert(1, os.path.join(os.path.dirname(os.path.realpath(__file__)), ".."))
@@ -189,7 +191,8 @@ def _create_access_scenario(self):
         ksu_cubesat_propagator.ephemeris_interval.set_implicit_interval(
             root.current_scenario.analysis_workbench_components.time_intervals.item("AnalysisInterval")
         )
-        ksu_cubesat_propagator.common_tasks.add_segments_from_online_source("47954")
+        tle_file = Path(__file__).parent.parent / "data" / "KSU-CUBESAT_47954_1Jun2020.tle"
+        ksu_cubesat_propagator.common_tasks.add_segments_from_file("47954", str(tle_file.resolve()))
         ksu_cubesat_propagator.automatic_update_enabled = True
         ksu_cubesat_propagator.propagate()
 

tests/doc_snippets_tests/stk_objects/satellite/satellite_snippets.py

@@ -21,6 +21,7 @@
 # SOFTWARE.
 
 import os
+import pytest
 import sys
 
 from ansys.stk.core.stkobjects import (
@@ -244,6 +245,7 @@ def SPICESatelliteSnippet(self, root, satellite):
         propagator.step = 60.0
         propagator.propagate()
 
+    @pytest.mark.skip(reason="Temporarily disabled due to connectivity.")
     def test_SGP4SatelliteSnippet(self):
         try:
             satellite = self.get_scenario().children.new(STKObjectType.SATELLITE, "satellite")

tests/extensions/data_analysis/graphs/data/iss_30Jun2015.tle

@@ -0,0 +1,2 @@
+1 25544U 98067A   15181.66666667  .00010560  00000-0  15157-3 0 00002
+2 25544 051.6456 016.7121 0003271 114.0602 085.9106 15.55446671950135

tests/extensions/data_analysis/graphs/data/iss_5Jun2022.tle

@@ -0,0 +1,2 @@
+1 25544U 98067A   22156.00000000  .00001944  00000-0  34115-4 0 00005
+2 25544 051.6451 036.9525 0004589 195.9534 179.2523 15.49879136343299

tests/extensions/data_analysis/graphs/test_access_graphs.py

@@ -27,6 +27,8 @@
 
 import matplotlib
 
+from pathlib import Path
+
 from ansys.stk.extensions.data_analysis.graphs.access_graphs import access_duration_pie_chart, bit_error_rate_line_chart, carrier_to_noise_ratio_line_chart, cumulative_dwell_cumulative_pie_chart, ebno_line_chart, radar_sar_azimuth_resolution_line_chart, radar_sar_time_resolution_line_chart, revisit_diagram_interval_pie_chart, aer_line_chart, access_interval_graph, az_el_polar_center_90_graph, angular_rates_line_chart, elevation_angle_line_chart, gaps_interval_graph, probability_of_detection_line_chart, radar_antenna_gain_line_chart, radar_propagation_loss_line_chart, radar_searchtrack_integration_line_chart, radar_searchtrack_snr_line_chart, radar_system_noise_line_chart, signal_to_noise_ratio_line_chart, sun_rfi_line_chart
 from stk_environment import stk_root
 
@@ -45,7 +47,8 @@ def basic_access(stk_root):
     satellite = scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     lighting_constraint = satellite.access_constraints.add_constraint(
@@ -73,7 +76,8 @@ def leap_second_access(stk_root):
     satellite = scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_30Jun2015.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     access = place.get_access_to_object(satellite)
@@ -125,7 +129,8 @@ def communications_access(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     facility = stk_root.current_scenario.children.new(STKObjectType.FACILITY, "Facility")
@@ -148,7 +153,8 @@ def sar_radar_access(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     facility = stk_root.current_scenario.children.new(STKObjectType.FACILITY, "Facility")

tests/extensions/data_analysis/graphs/test_chain_graphs.py

@@ -23,6 +23,7 @@
 """Test the `chain_graphs` module."""
 
 import pytest
+from pathlib import Path
 from ansys.stk.extensions.data_analysis.graphs.chain_graphs import complete_chain_access_interval_graph, individual_object_access_interval_graph, individual_strand_access_interval_graph, number_of_accesses_line_chart
 from stk_environment import stk_root
 
@@ -38,7 +39,8 @@ def chain(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     chain = scenario.children.new(STKObjectType.CHAIN, "Chain")

tests/extensions/data_analysis/graphs/test_comm_system_graphs.py

@@ -23,6 +23,7 @@
 """Test the `comm_system_graphs` module."""
 
 import pytest
+from pathlib import Path
 from ansys.stk.extensions.data_analysis.graphs.comm_system_graphs import carrier_to_noise_vs_time_line_chart
 from stk_environment import stk_root
 
@@ -38,7 +39,8 @@ def comm_system(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     comm_system =  scenario.children.new(STKObjectType.COMM_SYSTEM, "CommSystem")

tests/extensions/data_analysis/graphs/test_coverage_definition_graphs.py

@@ -24,6 +24,7 @@
 
 from ansys.stk.extensions.data_analysis.graphs.coverage_definition_graphs import coverage_by_latitude_line_chart, gap_duration_line_chart, gi_point_coverage_interval_graph, gi_point_prob_of_coverage_line_chart, gi_region_coverage_line_chart, gi_region_time_to_cover_line_chart, percent_coverage_line_chart, gi_region_full_coverage_interval_graph
 import pytest
+from pathlib import Path
 from stk_environment import stk_root
 
 @pytest.fixture()
@@ -38,7 +39,8 @@ def coverage_definition(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     coverage.asset_list.add("Satellite/Satellite")
@@ -64,7 +66,8 @@ def coverage_definition_selected_region(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     coverage.grid.bounds_type = CoverageBounds.CUSTOM_REGIONS

tests/extensions/data_analysis/graphs/test_figure_of_merit_graphs.py

@@ -24,6 +24,7 @@
 
 from ansys.stk.extensions.data_analysis.graphs.figure_of_merit_graphs import gi_region_fom_line_chart, gi_region_satisfaction_interval_graph, grid_stats_over_time_line_chart, satisfied_by_time_line_chart, value_by_latitude_line_chart, value_by_longitude_line_chart, gi_point_fom_line_chart, gi_point_satisfaction_interval_graph, gi_all_dop_line_chart
 import pytest
+from pathlib import Path
 from stk_environment import stk_root
 
 @pytest.fixture()
@@ -38,7 +39,8 @@ def fom(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     coverage.asset_list.add("Satellite/Satellite")
@@ -66,7 +68,8 @@ def fom_dop(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     coverage.asset_list.add("Satellite/Satellite")
@@ -98,7 +101,8 @@ def fom_selected_region(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     coverage.grid.bounds_type = CoverageBounds.CUSTOM_REGIONS

tests/extensions/data_analysis/graphs/test_satellite_graphs.py

@@ -24,6 +24,7 @@
 
 from ansys.stk.extensions.data_analysis.graphs.satellite_graphs import beta_angle_line_chart, classical_orbit_elements_line_chart, euler_angles_line_chart, fixed_position_velocity_line_chart, inertial_position_velocity_line_chart, j2000_position_velocity_line_chart, solar_elevation_body_fixed_line_chart, solar_intensity_line_chart, sun_vector_fixed_line_chart, yaw_pitch_roll_line_chart, sunlight_intervals_interval_pie_chart, cumulative_sunlight_cumulative_pie_chart, eclipse_times_interval_graph, lighting_times_interval_graph, solar_az_el_polar_center_0_graph, solar_aer_line_chart, lla_position_line_chart
 import pytest
+from pathlib import Path
 from stk_environment import stk_root
 
 @pytest.fixture()
@@ -37,7 +38,8 @@ def satellite(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     yield satellite

tests/extensions/data_analysis/graphs/test_sensor_graphs.py

@@ -22,6 +22,7 @@
 
 """Test the `sensor_graphs` module."""
 import pytest
+from pathlib import Path
 from stk_environment import stk_root
 from ansys.stk.extensions.data_analysis.graphs.sensor_graphs import azimuth_elevation_line_chart, footprint_area_line_chart
 
@@ -52,7 +53,8 @@ def sensor_from_satellite(stk_root):
     satellite = stk_root.current_scenario.children.new(STKObjectType.SATELLITE, "Satellite")
     satellite.set_propagator_type(PropagatorType.SGP4)
     propagator = satellite.propagator
-    propagator.common_tasks.add_segments_from_online_source("25544")
+    tle_file = Path(__file__).parent / "data" / "iss_5Jun2022.tle"
+    propagator.common_tasks.add_segments_from_file("25544", str(tle_file.resolve()))
     propagator.propagate()
 
     sensor = satellite.children.new(STKObjectType.SENSOR, "Sensor")

tests/generated/stk_tests/code_snippets/propagators/sgp4_snippets.py

@@ -20,6 +20,7 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
+import pytest
 from test_util import *
 from code_snippets.code_snippets_test_base import *
 from ansys.stk.core.stkobjects import *
@@ -83,6 +84,7 @@ def ConfigureSGP4WithFileSource(self, propagator: "PropagatorSGP4", tleFilePath:
         # Propagate
         propagator.propagate()
 
+    @pytest.mark.skip(reason="Temporarily disabled due to connectivity.")
     def ConfigureSGP4WithOnlineSource(self, propagator: "PropagatorSGP4"):
         # Configure time period
         propagator.ephemeris_interval.set_explicit_interval("1 Jan 2012 12:00:00.000", "2 Jan 2012 12:00:00.000")