zmc
2023-10-12 ed135d79df12a2466b52dae1a82326941211dcc9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import io
 
import pytest
 
import pandas as pd
from pandas.tests.extension.base.base import BaseExtensionTests
 
 
class BasePrintingTests(BaseExtensionTests):
    """Tests checking the formatting of your EA when printed."""
 
    @pytest.mark.parametrize("size", ["big", "small"])
    def test_array_repr(self, data, size):
        if size == "small":
            data = data[:5]
        else:
            data = type(data)._concat_same_type([data] * 5)
 
        result = repr(data)
        assert type(data).__name__ in result
        assert f"Length: {len(data)}" in result
        assert str(data.dtype) in result
        if size == "big":
            assert "..." in result
 
    def test_array_repr_unicode(self, data):
        result = str(data)
        assert isinstance(result, str)
 
    def test_series_repr(self, data):
        ser = pd.Series(data)
        assert data.dtype.name in repr(ser)
 
    def test_dataframe_repr(self, data):
        df = pd.DataFrame({"A": data})
        repr(df)
 
    def test_dtype_name_in_info(self, data):
        buf = io.StringIO()
        pd.DataFrame({"A": data}).info(buf=buf)
        result = buf.getvalue()
        assert data.dtype.name in result