fix pandas-related tests

h2oai · Nov 21, 2023 · 49e56f6 · 49e56f6
1 parent 8e0c6b1
commit 49e56f6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test-dt-stats.py b/tests/test-dt-stats.py
@@ -251,7 +251,7 @@ def test_dt_sd_special_cases(src, res):
 def test_dt_skew_simple():
     n = 12345
     DT = dt.Frame([0] * n + [1000] * (2*n) + [10000] * (3*n))
-    pd_skew = DT.to_pandas().skew()[0]
+    pd_skew = DT.to_pandas().skew().iloc[0]
     dt_skew = DT.skew1()
     assert isclose(pd_skew, dt_skew, rel_tol=1e-10)
 

diff --git a/tests/types/test-array.py b/tests/types/test-array.py
@@ -283,7 +283,7 @@ def test_arr32_to_and_from_pandas(pd):
     DT = dt.Frame(S=src)
     assert DT.type == dt.Type.arr32(dt.Type.int32)
     pdf = DT.to_pandas()
-    assert pdf.dtypes[0] == object
+    assert pdf.dtypes.iloc[0] == object
     assert pdf.columns.tolist() == ['S']
     assert pdf['S'].tolist() == src
     DT2 = dt.Frame(pdf)