fme

yt-project · Sep 12, 2020 · ee205cc · ee205cc
1 parent 73f1f34
commit ee205cc
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/yt/data_objects/tests/test_numpy_ops.py b/yt/data_objects/tests/test_numpy_ops.py
@@ -80,12 +80,14 @@ def test_mean_sum_integrate():
 
 
 def test_min_max():
+    fields = ("density", "temperature")
+    units = ("g/cm**3", "K")
     for nprocs in [-1, 1, 2, 16]:
         if nprocs == -1:
-            ds = fake_amr_ds(fields=("density", "temperature"), particles=20)
+            ds = fake_amr_ds(fields=fields, particles=20)
         else:
             ds = fake_random_ds(
-                32, nprocs=nprocs, fields=("density", "temperature"), particles=20
+                32, nprocs=nprocs, fields=fields, units=units, particles=20
             )
 
         ad = ds.all_data()
@@ -128,11 +130,13 @@ def test_min_max():
 
 
 def test_argmin():
+    fields = ("density", "temperature")
+    units = ("g/cm**3", "K")
     for nprocs in [-1, 1, 2, 16]:
         if nprocs == -1:
-            ds = fake_amr_ds(fields=("density", "temperature"))
+            ds = fake_amr_ds(fields=fields)
         else:
-            ds = fake_random_ds(32, nprocs=nprocs, fields=("density", "temperature"))
+            ds = fake_random_ds(32, nprocs=nprocs, fields=fields, units=units)
 
         ad = ds.all_data()
 
@@ -152,11 +156,13 @@ def test_argmin():
 
 
 def test_argmax():
+    fields = ("density", "temperature")
+    units = ("g/cm**3", "K")
     for nprocs in [-1, 1, 2, 16]:
         if nprocs == -1:
-            ds = fake_amr_ds(fields=("density", "temperature"))
+            ds = fake_amr_ds(fields=fields)
         else:
-            ds = fake_random_ds(32, nprocs=nprocs, fields=("density", "temperature"))
+            ds = fake_random_ds(32, nprocs=nprocs, fields=fields, units=units)
 
         ad = ds.all_data()
 

diff --git a/yt/data_objects/tests/test_spheres.py b/yt/data_objects/tests/test_spheres.py
@@ -91,7 +91,9 @@ def test_domain_sphere():
 
 
 def test_sphere_center():
-    ds = fake_random_ds(16, nprocs=8, fields=("density", "temperature", "velocity_x"))
+    fields = ("density", "temperature", "velocity_x")
+    units = ("g/cm**3", "K", "cm/s")
+    ds = fake_random_ds(16, nprocs=8, fields=fields, units=units)
 
     # Test if we obtain same center in different ways
     sp1 = ds.sphere("max", (0.25, "unitary"))