Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
pola-rs
GitHub Repository: pola-rs/polars
Path: blob/main/py-polars/tests/unit/test_conftest.py
6939 views
1
"""Tests for the testing infrastructure."""
2
3
import pytest
4
5
6
@pytest.mark.xfail
7
def test_memory_usage() -> None:
8
pytest.fail(reason="Disabled for now")
9
# """The ``memory_usage`` fixture gives somewhat accurate results."""
10
# memory_usage = memory_usage_without_pyarrow
11
# assert memory_usage.get_current() < 100_000
12
# assert memory_usage.get_peak() < 100_000
13
#
14
# # Memory from Python is tracked:
15
# b = b"X" * 1_300_000
16
# assert 1_300_000 <= memory_usage.get_current() <= 2_000_000
17
# assert 1_300_000 <= memory_usage.get_peak() <= 2_000_000
18
# del b
19
# assert memory_usage.get_current() <= 500_000
20
# assert 1_300_000 <= memory_usage.get_peak() <= 2_000_000
21
# memory_usage.reset_tracking()
22
# assert memory_usage.get_current() < 100_000
23
# assert memory_usage.get_peak() < 100_000
24
#
25
# # Memory from Polars is tracked:
26
# df = pl.DataFrame({"x": pl.arange(0, 1_000_000, eager=True, dtype=pl.Int64)})
27
# del df
28
# peak_bytes = memory_usage.get_peak()
29
# assert 8_000_000 <= peak_bytes < 8_500_000
30
#
31
# memory_usage.reset_tracking()
32
# assert memory_usage.get_peak() < 1_000_000
33
#
34
# # Memory from NumPy is tracked:
35
# arr = np.ones((1_400_000,), dtype=np.uint8)
36
# del arr
37
# peak = memory_usage.get_peak()
38
# assert 1_400_000 < peak < 1_500_000
39
40