def test_series(self):
pandas = pytest.importorskip("pandas")
df = pandas.DataFrame(
data=[["Bob", 20, "Apprentice"], ["Alice", 23, "Secretary"],
["Mario", 39, "Plumber"]],
columns=["Name", "Age", "Occupation"],
)
ser = df.iloc[0]
assert head(ser) == """\
Name Bob
Age 20
Occupation Apprentice\
"""
assert head(ser, 1) == "Name Bob"
def test_data_frame(self):
pandas = pytest.importorskip("pandas")
df = pandas.DataFrame(
data=[["Bob", 20, "Apprentice"], ["Alice", 23, "Secretary"],
["Mario", 39, "Plumber"]],
columns=["Name", "Age", "Occupation"],
)
assert head(df) == """ Name Age Occupation
0 Bob 20 Apprentice
1 Alice 23 Secretary
2 Mario 39 Plumber\
"""
assert head(df,
1) == " Name Age Occupation\n0 Bob 20 Apprentice"
def test_str(self):
assert head("Hello World") == "Hello Worl..."
assert head("Hello World", 11) == "Hello World"
assert head("Hello World", 5) == "Hello..."
def test_list(self, args, expects):
assert head(*args) == expects
def test_tuple(self, args, expects):
assert head(*args) == expects