Update test_decorator.py

pyiron · Oct 30, 2024 · 43bfd3d · 43bfd3d
1 parent 32d275b
commit 43bfd3d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/unit/flex/test_decorator.py b/tests/unit/flex/test_decorator.py
@@ -1,5 +1,5 @@
 from pyiron_base._tests import TestWithProject
-from pyiron_base import pyiron_job
+from pyiron_base import job
 import unittest
 
 
@@ -8,11 +8,11 @@ def tearDown(self):
         self.project.remove_jobs(recursive=True, silently=True)
 
     def test_delayed(self):
-        @pyiron_job()
+        @job()
         def my_function_a(a, b=8):
             return a + b
 
-        @pyiron_job(cores=2)
+        @job(cores=2)
         def my_function_b(a, b=8):
             return a + b
 
@@ -24,11 +24,11 @@ def my_function_b(a, b=8):
         self.assertEqual(len(edges_lst), 6)
 
     def test_delayed_simple(self):
-        @pyiron_job
+        @job
         def my_function_a(a, b=8):
             return a + b
 
-        @pyiron_job
+        @job
         def my_function_b(a, b=8):
             return a + b