MNT: cleanup deprecated internal imports (6/n)

yt-project · May 3, 2023 · afb2166 · afb2166
1 parent 5ca8421
commit afb2166
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 9 deletions.
diff --git a/yt/data_objects/tests/test_data_containers.py b/yt/data_objects/tests/test_data_containers.py
@@ -5,12 +5,11 @@
 
 import numpy as np
 from nose.tools import assert_raises
-from numpy.testing import assert_array_equal
+from numpy.testing import assert_array_equal, assert_equal
 
 from yt.data_objects.data_containers import YTDataContainer
 from yt.data_objects.particle_filters import particle_filter
 from yt.testing import (
-    assert_equal,
     fake_amr_ds,
     fake_particle_ds,
     fake_random_ds,

diff --git a/yt/data_objects/tests/test_regions.py b/yt/data_objects/tests/test_regions.py
@@ -1,4 +1,5 @@
-from yt.testing import assert_array_equal, assert_raises, fake_amr_ds, fake_random_ds
+from numpy.testing import assert_array_equal, assert_raises, fake_amr_ds, fake_random_ds
+
 from yt.units import cm
 
 

diff --git a/yt/fields/tests/test_particle_fields.py b/yt/fields/tests/test_particle_fields.py
@@ -1,6 +1,4 @@
-from numpy.testing import assert_allclose_units
-
-from yt.testing import requires_file, requires_module
+from yt.testing import assert_allclose_units, requires_file, requires_module
 from yt.utilities.answer_testing.framework import data_dir_load
 
 g30 = "IsolatedGalaxy/galaxy0030/galaxy0030"

diff --git a/yt/frontends/amrvac/tests/test_units_override.py b/yt/frontends/amrvac/tests/test_units_override.py
@@ -1,6 +1,6 @@
-from numpy.testing import assert_allclose_units, assert_raises
+from numpy.testing import assert_raises
 
-from yt.testing import requires_file
+from yt.testing import assert_allclose_units, requires_file
 from yt.units import YTQuantity
 from yt.utilities.answer_testing.framework import data_dir_load
 

diff --git a/yt/frontends/ytdata/tests/test_outputs.py b/yt/frontends/ytdata/tests/test_outputs.py
@@ -3,6 +3,7 @@
 import tempfile
 
 import numpy as np
+from numpy.testing import assert_array_equal, assert_equal
 
 from yt.data_objects.api import create_profile
 from yt.frontends.ytdata.api import (
@@ -14,7 +15,7 @@
     save_as_dataset,
 )
 from yt.loaders import load
-from yt.testing import assert_allclose_units, assert_array_equal, assert_equal
+from yt.testing import assert_allclose_units
 from yt.units.yt_array import YTArray, YTQuantity
 from yt.utilities.answer_testing.framework import (
     AnswerTestingTest,