|
15 | 15 | from pandas import ( |
16 | 16 | CategoricalIndex, |
17 | 17 | DataFrame, |
| 18 | + Index, |
18 | 19 | MultiIndex, |
19 | 20 | Series, |
20 | 21 | date_range, |
@@ -360,7 +361,7 @@ def test_info_memory_usage(): |
360 | 361 | df = DataFrame(data) |
361 | 362 | df.columns = dtypes |
362 | 363 |
|
363 | | - df_with_object_index = DataFrame({"a": [1]}, index=["foo"]) |
| 364 | + df_with_object_index = DataFrame({"a": [1]}, index=Index(["foo"], dtype=object)) |
364 | 365 | df_with_object_index.info(buf=buf, memory_usage=True) |
365 | 366 | res = buf.getvalue().splitlines() |
366 | 367 | assert re.match(r"memory usage: [^+]+\+", res[-1]) |
@@ -398,25 +399,25 @@ def test_info_memory_usage(): |
398 | 399 |
|
399 | 400 | @pytest.mark.skipif(PYPY, reason="on PyPy deep=True doesn't change result") |
400 | 401 | def test_info_memory_usage_deep_not_pypy(): |
401 | | - df_with_object_index = DataFrame({"a": [1]}, index=["foo"]) |
| 402 | + df_with_object_index = DataFrame({"a": [1]}, index=Index(["foo"], dtype=object)) |
402 | 403 | assert ( |
403 | 404 | df_with_object_index.memory_usage(index=True, deep=True).sum() |
404 | 405 | > df_with_object_index.memory_usage(index=True).sum() |
405 | 406 | ) |
406 | 407 |
|
407 | | - df_object = DataFrame({"a": ["a"]}) |
| 408 | + df_object = DataFrame({"a": Series(["a"], dtype=object)}) |
408 | 409 | assert df_object.memory_usage(deep=True).sum() > df_object.memory_usage().sum() |
409 | 410 |
|
410 | 411 |
|
411 | 412 | @pytest.mark.xfail(not PYPY, reason="on PyPy deep=True does not change result") |
412 | 413 | def test_info_memory_usage_deep_pypy(): |
413 | | - df_with_object_index = DataFrame({"a": [1]}, index=["foo"]) |
| 414 | + df_with_object_index = DataFrame({"a": [1]}, index=Index(["foo"], dtype=object)) |
414 | 415 | assert ( |
415 | 416 | df_with_object_index.memory_usage(index=True, deep=True).sum() |
416 | 417 | == df_with_object_index.memory_usage(index=True).sum() |
417 | 418 | ) |
418 | 419 |
|
419 | | - df_object = DataFrame({"a": ["a"]}) |
| 420 | + df_object = DataFrame({"a": Series(["a"], dtype=object)}) |
420 | 421 | assert df_object.memory_usage(deep=True).sum() == df_object.memory_usage().sum() |
421 | 422 |
|
422 | 423 |
|
|
0 commit comments