MultiFab.__len__()

We already have `.size` and we can iterate the `FabArray`, so we should support `len(mf)`, too.
AMReX-Codes · Jan 25, 2025 · f0bd5c6 · f0bd5c6
1 parent 41f6952
commit f0bd5c6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/Base/MultiFab.cpp b/src/Base/MultiFab.cpp
@@ -143,6 +143,8 @@ void init_MultiFab(py::module &m)
             "Return number of variables (aka components) associated with each point.")
         .def_property_readonly("size", &FabArrayBase::size,
             "Return the number of FABs in the FabArray.")
+        .def("__len__", &FabArrayBase::size,
+            "Return the number of FABs in the FabArray.")
 
         .def_property_readonly("n_grow_vect", &FabArrayBase::nGrowVect,
             "Return the grow factor (per direction) that defines the region of definition.")

diff --git a/tests/test_multifab.py b/tests/test_multifab.py
@@ -269,6 +269,8 @@ def test_mfab_ops(boxarr, distmap, nghost):
 
 
 def test_mfab_mfiter(mfab):
+    assert len(mfab) == 8
+
     assert iter(mfab).is_valid
     assert iter(mfab).length == 8