Do not export multi type support to python -> use sys.getsizeof

Author: schnellerhase

python/dolfinx/wrappers/common.cpp

@@ -146,20 +146,20 @@ void common(nb::module_& m)
   m.attr("ufcx_signature") = dolfinx::ufcx_signature();
   m.attr("version") = dolfinx::version();
 
-  enum MemoryUnit : std::int64_t
-  {
-    byte = dolfinx::common::byte.value,
-    kilobyte = dolfinx::common::kilobyte.value,
-    megabyte = dolfinx::common::megabyte.value,
-    gigabyte = dolfinx::common::gigabyte.value,
-    terabyte = dolfinx::common::terabyte.value,
-  };
-  nb::enum_<MemoryUnit>(m, "MemoryUnit")
-      .value("byte", MemoryUnit::byte)
-      .value("kilobyte", MemoryUnit::kilobyte)
-      .value("megabyte", MemoryUnit::megabyte)
-      .value("gigabyte", MemoryUnit::gigabyte)
-      .value("terabyte", MemoryUnit::terabyte);
+  // enum MemoryUnit : std::int64_t
+  // {
+  //   byte = dolfinx::common::byte.value,
+  //   kilobyte = dolfinx::common::kilobyte.value,
+  //   megabyte = dolfinx::common::megabyte.value,
+  //   gigabyte = dolfinx::common::gigabyte.value,
+  //   terabyte = dolfinx::common::terabyte.value,
+  // };
+  // nb::enum_<MemoryUnit>(m, "MemoryUnit")
+  //     .value("byte", MemoryUnit::byte)
+  //     .value("kilobyte", MemoryUnit::kilobyte)
+  //     .value("megabyte", MemoryUnit::megabyte)
+  //     .value("gigabyte", MemoryUnit::gigabyte)
+  //     .value("terabyte", MemoryUnit::terabyte);
 
   nb::enum_<dolfinx::Table::Reduction>(m, "Reduction")
       .value("max", dolfinx::Table::Reduction::max)
@@ -265,25 +265,27 @@ void common(nb::module_& m)
             return dolfinx_wrappers::as_nbarray(std::move(local));
           },
           nb::arg("global"))
-      .def("memory",
-           [](const dolfinx::common::IndexMap& self, MemoryUnit unit)
-           {
-             switch (unit)
-             {
-             case MemoryUnit::byte:
-               return double(
-                   dolfinx::common::memory(self, dolfinx::common::byte));
-             case MemoryUnit::kilobyte:
-               return dolfinx::common::memory(self, dolfinx::common::kilobyte);
-             case MemoryUnit::megabyte:
-               return dolfinx::common::memory(self, dolfinx::common::megabyte);
-             case MemoryUnit::gigabyte:
-               return dolfinx::common::memory(self, dolfinx::common::gigabyte);
-             case MemoryUnit::terabyte:
-               return dolfinx::common::memory(self, dolfinx::common::terabyte);
-             }
-             throw std::runtime_error("Not a valid memory unit.");
-           });
+      .def("__sizeof__", [](const dolfinx::common::IndexMap& self)
+           { return dolfinx::common::memory(self, dolfinx::common::byte); });
+  // .def("memory",
+  //      [](const dolfinx::common::IndexMap& self, MemoryUnit unit)
+  //      {
+  //        switch (unit)
+  //        {
+  //        case MemoryUnit::byte:
+  //          return double(
+  //              dolfinx::common::memory(self, dolfinx::common::byte));
+  //        case MemoryUnit::kilobyte:
+  //          return dolfinx::common::memory(self, dolfinx::common::kilobyte);
+  //        case MemoryUnit::megabyte:
+  //          return dolfinx::common::memory(self, dolfinx::common::megabyte);
+  //        case MemoryUnit::gigabyte:
+  //          return dolfinx::common::memory(self, dolfinx::common::gigabyte);
+  //        case MemoryUnit::terabyte:
+  //          return dolfinx::common::memory(self, dolfinx::common::terabyte);
+  //        }
+  //        throw std::runtime_error("Not a valid memory unit.");
+  //      });
 
   // dolfinx::common::Timer
   nb::class_<dolfinx::common::Timer<std::chrono::high_resolution_clock>>(

python/test/unit/common/test_index_map.py

@@ -5,6 +5,7 @@
 # SPDX-License-Identifier:    LGPL-3.0-or-later
 
 import math
+import sys
 
 from mpi4py import MPI
 
@@ -42,8 +43,9 @@ def test_sub_index_map():
         comm, map_local_size, [dest_ranks, src_ranks], map_ghosts, src_ranks
     )
     assert map.size_global == map_local_size * comm.size
-    assert map.memory(dolfinx.cpp.common.MemoryUnit.byte) > 0
-    assert map.memory(dolfinx.cpp.common.MemoryUnit.kilobyte) > 0
+    assert sys.getsizeof(map) > 0
+    # assert map.memory(dolfinx.cpp.common.MemoryUnit.byte) > 0
+    # assert map.memory(dolfinx.cpp.common.MemoryUnit.kilobyte) > 0
 
     # Build list for each rank of the first (myrank + myrank % 2) local
     # indices