generalized test_measure_format_f

eli-b · Sep 30, 2022 · a8108c9 · a8108c9
1 parent 1958d2a
commit a8108c9
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/static_frame/test/unit/test_memory_measure.py b/static_frame/test/unit/test_memory_measure.py
@@ -418,19 +418,23 @@ def test_measure_format_f(self) -> None:
         a1 = np.array((1, 2), dtype=np.int64)
         a2 = a1[:]
 
+        mempty = MaterializedArray(empty, format=MeasureFormat.LOCAL_MATERIALIZED)
+        ma1 = MaterializedArray(a1, format=MeasureFormat.LOCAL_MATERIALIZED)
+        ma2 = MaterializedArray(a2, format=MeasureFormat.LOCAL_MATERIALIZED)
+
         self.assertEqual(memory_total(empty,
                 format=MeasureFormat.LOCAL_MATERIALIZED),
-                getsizeof(None) + getsizeof(empty),
+                getsizeof(mempty),
                 )
 
         self.assertEqual(memory_total(a1,
                 format=MeasureFormat.LOCAL_MATERIALIZED),
-                getsizeof(None) + getsizeof(a1),
+                getsizeof(ma1),
                 )
 
         self.assertEqual(memory_total(a2,
                 format=MeasureFormat.LOCAL_MATERIALIZED),
-                getsizeof(None) + getsizeof(a2),
+                getsizeof(ma2),
                 )
 
     #---------------------------------------------------------------------------