Skip to content

Commit 1f91170

Browse files
GH1217 PR feedback
1 parent 5a7fe40 commit 1f91170

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

tests/test_frame.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -629,7 +629,22 @@ def test_types_itertuples() -> None:
629629
)
630630

631631
for t1 in df.itertuples():
632-
check(assert_type(t1, _PandasNamedTuple), _PandasNamedTuple)
632+
assert_type(t1, _PandasNamedTuple)
633+
assert t1.__class__.__name__ == "Pandas"
634+
assert isinstance(t1.Index, int)
635+
assert isinstance(t1.col1, int)
636+
assert isinstance(t1.col2, int)
637+
for k in [0, 1]:
638+
assert isinstance(t1[k], int)
639+
640+
for t1 in df.itertuples(name="FooBar"):
641+
assert_type(t1, _PandasNamedTuple)
642+
assert t1.__class__.__name__ == "FooBar"
643+
assert isinstance(t1.Index, int)
644+
assert isinstance(t1.col1, int)
645+
assert isinstance(t1.col2, int)
646+
for k in [0, 1]:
647+
assert isinstance(t1[k], int)
633648

634649

635650
def test_types_items() -> None:

0 commit comments

Comments
 (0)