Merge pull request #4964 from chrishavlin/tipsy_bbox_hsml

yt-project · Aug 17, 2024 · ba832f4 · ba832f4
2 parents 981a8be + 0a3cefc
commit ba832f4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/yt/frontends/tipsy/io.py b/yt/frontends/tipsy/io.py
@@ -58,7 +58,10 @@ def _fill_fields(self, fields, vals, hsml, mask, data_file):
         if mask is None:
             size = 0
         elif isinstance(mask, slice):
-            size = vals[fields[0]].size
+            if fields[0] == "smoothing_length":
+                size = hsml.size
+            else:
+                size = vals[fields[0]].size
         else:
             size = mask.sum()
         rv = {}

diff --git a/yt/frontends/tipsy/tests/test_outputs.py b/yt/frontends/tipsy/tests/test_outputs.py
@@ -112,3 +112,9 @@ def test_tipsy_index():
     ds = data_dir_load(tipsy_gal)
     sl = ds.slice("z", 0.0)
     assert sl["gas", "density"].shape[0] != 0
+
+
+@requires_file(tipsy_gal)
+def test_tipsy_smoothing_length():
+    ds = data_dir_load(tipsy_gal)
+    _ = ds.all_data()["Gas", "smoothing_length"]