#6022 - fmt

radiasoft · Aug 10, 2023 · 9dac7fe · 9dac7fe
1 parent d695f0d
commit 9dac7fe
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/template/srw_resize_3d_test.py b/tests/template/srw_resize_3d_test.py
@@ -4,20 +4,22 @@
 :copyright: Copyright (c) 2023 RadiaSoft LLC.  All Rights Reserved.
 :license: http://www.apache.org/licenses/LICENSE-2.0.html
 """
-from pykern.pkcollections import PKDict
-from pykern.pkunit import pkeq, pkok
+import pytest
 
 _N_BINS = 100
 
 
 def test_srw_resize_3d():
     from pykern import pkconfig
+
     pkconfig.reset_state_for_testing(
         dict(
             SIREPO_JOB_MAX_MESSAGE_BYTES=str(_N_BINS * _N_BINS),
         )
     )
 
+    from pykern.pkcollections import PKDict
+    from pykern import pkunit
     from sirepo import job
     from sirepo.template import srw
     import numpy
@@ -27,12 +29,12 @@ def test_srw_resize_3d():
         [0, 0, 0, 0.0, 1.0, _N_BINS, 0.0, 1.0, _N_BINS],
         PKDict(),
     )
-    pkok(
+    pkunit.pkok(
         xr[2] < _N_BINS,
         "did not reduce bins nx={}",
         xr[2],
     )
-    pkok(
+    pkunit.pkok(
         yr[2] < _N_BINS,
         "did not reduce bins ny={}",
         yr[2],