Fix tests

nomad-coe · Feb 20, 2024 · a33caa5 · a33caa5
1 parent d6ecf98
commit a33caa5
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/tests/test_runschema.py b/tests/test_runschema.py
@@ -16,12 +16,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import runschema
+from ..runschema.run import Run
+from ..runschema.calculation import Calculation
+from ..runschema.method import Method
+from ..runschema.system import System
 
 
 def test_schema():
-    run = runschema.run.Run()
+    run = Run()
     assert run.m_def.definition_id == 'fc17fdfdcba5aef2c02835a81ed65f95f314b687'
-    run.calculation.append(runschema.calculation.Calculation())
-    run.method.append(runschema.method.Method())
-    run.system.append(runschema.system.System())
+    run.calculation.append(Calculation())
+    run.method.append(Method())
+    run.system.append(System())