MCPcopy Create free account
hub / github.com/dask/dask / test_memory_usage

Function test_memory_usage

dask/dataframe/dask_expr/tests/test_collection.py:504–515  ·  view source on GitHub ↗
(pdf)

Source from the content-addressed store, hash-verified

502
503
504def test_memory_usage(pdf):
505 # Results are not equal with RangeIndex because pandas has one RangeIndex while
506 # we have one RangeIndex per partition
507 pdf.index = np.arange(len(pdf))
508 df = from_pandas(pdf)
509 assert_eq(df.memory_usage(), pdf.memory_usage())
510 assert_eq(df.memory_usage(index=False), pdf.memory_usage(index=False))
511 assert_eq(df.x.memory_usage(), pdf.x.memory_usage())
512 assert_eq(df.x.memory_usage(index=False), pdf.x.memory_usage(index=False))
513 assert_eq(df.index.memory_usage(), pdf.index.memory_usage())
514 with pytest.raises(TypeError, match="got an unexpected keyword"):
515 df.index.memory_usage(index=True)
516
517
518@pytest.mark.parametrize("func", [M.nlargest, M.nsmallest])

Callers

nothing calls this directly

Calls 4

from_pandasFunction · 0.90
assert_eqFunction · 0.90
arangeMethod · 0.45
memory_usageMethod · 0.45

Tested by

no test coverage detected