Python Packaging Updates (#3057)

Updating geosPythonPackages --------- Co-authored-by: Christopher Sherman <sherman27@llnl.gov>
GEOS-DEV · Jun 21, 2024 · f91800b · f91800b
1 parent d6b6e0b
commit f91800b
Show file tree

Hide file tree

Showing 55 changed files with 68 additions and 159 deletions.
diff --git a/inputFiles/almContactMechanics/alm.ats b/inputFiles/almContactMechanics/alm.ats
@@ -1,6 +1,5 @@
 import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 2.0E-4

diff --git a/inputFiles/compositionalMultiphaseFlow/benchmarks/SPE10/SPE10.ats b/inputFiles/compositionalMultiphaseFlow/benchmarks/SPE10/SPE10.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-6

diff --git a/.../compositionalMultiphaseFlow/benchmarks/buckleyLeverettProblem/buckleyLeverettProblem.ats b/.../compositionalMultiphaseFlow/benchmarks/buckleyLeverettProblem/buckleyLeverettProblem.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-6

diff --git a/...tFiles/compositionalMultiphaseFlow/benchmarks/isothermalLeakyWell/isothermalLeakyWell.ats b/...tFiles/compositionalMultiphaseFlow/benchmarks/isothermalLeakyWell/isothermalLeakyWell.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-6

diff --git a/inputFiles/compositionalMultiphaseFlow/benchmarks/thermalLeakyWell/thermalLeakyWell.ats b/inputFiles/compositionalMultiphaseFlow/benchmarks/thermalLeakyWell/thermalLeakyWell.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-6

diff --git a/inputFiles/compositionalMultiphaseFlow/c1-ppu/c1_ppu.ats b/inputFiles/compositionalMultiphaseFlow/c1-ppu/c1_ppu.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-6

diff --git a/inputFiles/compositionalMultiphaseFlow/compositionalMultiphaseFlow.ats b/inputFiles/compositionalMultiphaseFlow/compositionalMultiphaseFlow.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-6

diff --git a/inputFiles/compositionalMultiphaseWell/benchmarks/Class09Pb3/Class09Pb3.ats b/inputFiles/compositionalMultiphaseWell/benchmarks/Class09Pb3/Class09Pb3.ats
@@ -1,15 +1,9 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-6
 restartcheck_params["rtol"] = 1.0E-5
 
-# exclude any derivative fields from restart checks
-# example field name: dGlobalComponentFraction_dPressure
-#restartcheck_params["exclude"] = [r"d[A-Z]\w+_d[A-Z]\w+"]
-
 decks = [
     TestDeck(
         name="class09_pb3_smoke_3d",

diff --git a/inputFiles/compositionalMultiphaseWell/benchmarks/Egg/compositionalMultiphaseWell_Egg.ats b/inputFiles/compositionalMultiphaseWell/benchmarks/Egg/compositionalMultiphaseWell_Egg.ats
@@ -1,15 +1,9 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-6
 restartcheck_params["rtol"] = 1.0E-5
 
-# exclude any derivative fields from restart checks
-# example field name: dGlobalComponentFraction_dPressure
-#restartcheck_params["exclude"] = [r"d[A-Z]\w+_d[A-Z]\w+"]
-
 decks = [
     TestDeck(
         name="deadOilEgg_smoke_3d",

diff --git a/inputFiles/compositionalMultiphaseWell/compositionalMultiphaseWell.ats b/inputFiles/compositionalMultiphaseWell/compositionalMultiphaseWell.ats
@@ -1,15 +1,9 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-6
 restartcheck_params["rtol"] = 1.0E-5
 
-# exclude any derivative fields from restart checks
-# example field name: dGlobalComponentFraction_dPressure
-#restartcheck_params["exclude"] = [r"d[A-Z]\w+_d[A-Z]\w+"]
-
 decks = [
     TestDeck(
         name="compositional_multiphase_wells_1d",

diff --git a/inputFiles/efemFractureMechanics/efemFractureMechanics.ats b/inputFiles/efemFractureMechanics/efemFractureMechanics.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, CurveCheckParameters, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, CurveCheckParameters, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {'atol': 1e-08, 'rtol': 4e-07}
 

diff --git a/inputFiles/hydraulicFracturing/Hydrofracturing.ats b/inputFiles/hydraulicFracturing/Hydrofracturing.ats
@@ -1,6 +1,5 @@
 import copy
-import geos_ats
-from geos_ats.test_builder import generate_geos_tests, RestartcheckParameters, CurveCheckParameters, TestDeck
+from geos.ats.test_builder import generate_geos_tests, RestartcheckParameters, CurveCheckParameters, TestDeck
 
 # Define cases
 kgd_viscosity_dominated = {

diff --git a/inputFiles/inducedSeismicity/inducedSeismicity.ats b/inputFiles/inducedSeismicity/inducedSeismicity.ats
@@ -1,6 +1,5 @@
 import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests, CurveCheckParameters
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests, CurveCheckParameters
 
 curvecheck_params = {}
 curvecheck_params["filename"] = "seismicityRate.hdf5"

diff --git a/inputFiles/lagrangianContactMechanics/contactMechanics.ats b/inputFiles/lagrangianContactMechanics/contactMechanics.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 2.0E-4

diff --git a/inputFiles/main.ats b/inputFiles/main.ats
@@ -1,7 +1,6 @@
 """
 Primary ATS script
 """
-import os
 import ats
 from pathlib import Path
 

diff --git a/inputFiles/materialPointMethod/dfgMovingGrid/solidMechanicsMPM_dfgMovingGrid.ats b/inputFiles/materialPointMethod/dfgMovingGrid/solidMechanicsMPM_dfgMovingGrid.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1.0E-8

diff --git a/inputFiles/materialPointMethod/particleRepartition/solidMechanicsMPM_particleRepartition.ats b/inputFiles/materialPointMethod/particleRepartition/solidMechanicsMPM_particleRepartition.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1.0E-8

diff --git a/inputFiles/materialPointMethod/singleParticle/solidMechanicsMPM_singleParticle.ats b/inputFiles/materialPointMethod/singleParticle/solidMechanicsMPM_singleParticle.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1.0E-8

diff --git a/inputFiles/multiphaseFlowFractures/multiphaseFlowFractures.ats b/inputFiles/multiphaseFlowFractures/multiphaseFlowFractures.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-10

diff --git a/inputFiles/phaseField/PhaseFieldFracture.ats b/inputFiles/phaseField/PhaseFieldFracture.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-06

diff --git a/...les/poromechanics/nonlinearAcceleration/validationCase/poroelasticCoupling_validation.ats b/...les/poromechanics/nonlinearAcceleration/validationCase/poroelasticCoupling_validation.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 
 class Description(object):

diff --git a/inputFiles/poromechanics/poroElasticCoupling.ats b/inputFiles/poromechanics/poroElasticCoupling.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 
 class Description(object):

diff --git a/inputFiles/poromechanics/viscoPlastic.ats b/inputFiles/poromechanics/viscoPlastic.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck
+from geos.ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-5

diff --git a/inputFiles/poromechanicsFractures/poromechanicsFractures.ats b/inputFiles/poromechanicsFractures/poromechanicsFractures.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {'atol': 1e-07, 'rtol': 4e-06}
 

diff --git a/inputFiles/proppant/proppantTransport.ats b/inputFiles/proppant/proppantTransport.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-10

diff --git a/inputFiles/simplePDE/SimpleSolvers.ats b/inputFiles/simplePDE/SimpleSolvers.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-08

diff --git a/inputFiles/singlePhaseFlow/singlePhaseFlow.ats b/inputFiles/singlePhaseFlow/singlePhaseFlow.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-10

diff --git a/inputFiles/singlePhaseFlow/thermalSinglePhaseFlow.ats b/inputFiles/singlePhaseFlow/thermalSinglePhaseFlow.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-06

diff --git a/inputFiles/singlePhaseFlowFractures/singlePhaseFlowFractures.ats b/inputFiles/singlePhaseFlowFractures/singlePhaseFlowFractures.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-10

diff --git a/inputFiles/singlePhaseWell/singlePhaseWell.ats b/inputFiles/singlePhaseWell/singlePhaseWell.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-09

diff --git a/inputFiles/solidMechanics/SSLE.ats b/inputFiles/solidMechanics/SSLE.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck
+from geos.ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-10

diff --git a/inputFiles/solidMechanics/anisotropic.ats b/inputFiles/solidMechanics/anisotropic.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import generate_geos_tests, TestDeck, RestartcheckParameters
+from geos.ats.test_builder import generate_geos_tests, TestDeck, RestartcheckParameters
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-5

diff --git a/inputFiles/solidMechanics/beamBending.ats b/inputFiles/solidMechanics/beamBending.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import RestartcheckParameters, CurveCheckParameters, TestDeck, generate_geos_tests
+from geos.ats.test_builder import RestartcheckParameters, CurveCheckParameters, TestDeck, generate_geos_tests
 
 restartcheck_params = RestartcheckParameters(atol=1.0E-3, rtol=1.0E-7)
 

diff --git a/inputFiles/solidMechanics/gravity.ats b/inputFiles/solidMechanics/gravity.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import generate_geos_tests, TestDeck, RestartcheckParameters
+from geos.ats.test_builder import generate_geos_tests, TestDeck, RestartcheckParameters
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-5

diff --git a/inputFiles/solidMechanics/plasticCubeReset.ats b/inputFiles/solidMechanics/plasticCubeReset.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck
+from geos.ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-5

diff --git a/inputFiles/solidMechanics/plasticWellbore.ats b/inputFiles/solidMechanics/plasticWellbore.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-06

diff --git a/inputFiles/solidMechanics/sedov.ats b/inputFiles/solidMechanics/sedov.ats
@@ -1,6 +1,5 @@
 # Integrated Test Docs Begin Parameters
-import geos_ats
-from geos_ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck, CurveCheckParameters
+from geos.ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck, CurveCheckParameters
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 2.0E-10

diff --git a/inputFiles/solidMechanics/viscoPlastic.ats b/inputFiles/solidMechanics/viscoPlastic.ats
@@ -1,5 +1,4 @@
-import geos_ats
-from geos_ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck
+from geos.ats.test_builder import generate_geos_tests, RestartcheckParameters, TestDeck
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-5

diff --git a/inputFiles/solidMechanics/wellbore.ats b/inputFiles/solidMechanics/wellbore.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 0.0001

diff --git a/inputFiles/surfaceGeneration/SurfaceGenerator.ats b/inputFiles/surfaceGeneration/SurfaceGenerator.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params["atol"] = 1.0E-5

diff --git a/inputFiles/thermalMultiphaseFlow/thermalMultiphaseFlow.ats b/inputFiles/thermalMultiphaseFlow/thermalMultiphaseFlow.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-06

diff --git a/inputFiles/thermalSinglePhaseFlowFractures/thermalSinglePhaseFlowFractures.ats b/inputFiles/thermalSinglePhaseFlowFractures/thermalSinglePhaseFlowFractures.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 1e-06

diff --git a/inputFiles/thermoPoromechanics/thermoPoromechanics.ats b/inputFiles/thermoPoromechanics/thermoPoromechanics.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {'atol': 1e-06, 'rtol': 4e-06}
 

diff --git a/inputFiles/thermoPoromechanicsFractures/thermoPoromechanicsFractures.ats b/inputFiles/thermoPoromechanicsFractures/thermoPoromechanicsFractures.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {'atol': 1e-06, 'rtol': 4e-06}
 

diff --git a/inputFiles/wavePropagation/AcousticElasticSEM.ats b/inputFiles/wavePropagation/AcousticElasticSEM.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 0.0001

diff --git a/inputFiles/wavePropagation/AcousticFirstOrderSEM.ats b/inputFiles/wavePropagation/AcousticFirstOrderSEM.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 0.0001

diff --git a/inputFiles/wavePropagation/AcousticSEM.ats b/inputFiles/wavePropagation/AcousticSEM.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 0.0001

diff --git a/inputFiles/wavePropagation/ElasticFirstOrderSEM.ats b/inputFiles/wavePropagation/ElasticFirstOrderSEM.ats
@@ -1,6 +1,4 @@
-import os
-import geos_ats
-from geos_ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
+from geos.ats.test_builder import TestDeck, RestartcheckParameters, generate_geos_tests
 
 restartcheck_params = {}
 restartcheck_params['atol'] = 0.0001