Fixed Pandas warnings in test_quad.py (#738)

QuantEcon · Aug 30, 2024 · c2ed3d6 · c2ed3d6
1 parent bcba2ac
commit c2ed3d6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/quantecon/tests/test_quad.py b/quantecon/tests/test_quad.py
@@ -114,7 +114,7 @@ def setup_method(cls):
             for kind in kinds:
                 for num in n:
                     num_in = num ** 2 if len(kind) == 1 else num
-                    quad_rect_res1d.loc[func_name, num][kind] = \
+                    quad_rect_res1d.loc[(func_name, num), kind] = \
                         quadrect(func, num_in, a, b, kind,
                                  random_state=random_state)
 
@@ -134,10 +134,10 @@ def setup_method(cls):
 
         for num in n:
             for kind in kinds2[:4]:
-                data2.loc[num**2][kind] = quadrect(f1_2, [num, num],
+                data2.loc[num**2, kind] = quadrect(f1_2, [num, num],
                                                   a[0], b[0], kind)
             for kind in kinds2[4:]:
-                data2.loc[num**2][kind] = quadrect(f1_2, num**2, a[0],
+                data2.loc[num**2, kind] = quadrect(f1_2, num**2, a[0],
                                                   b[0], kind)
 
         cls.data2d1 = data2
@@ -149,7 +149,7 @@ def setup_method(cls):
 
         for num in n3:
             for kind in kinds2[3:]:
-                data3.loc[num][kind] = quadrect(f2_2, num, a[1], b[1], kind)
+                data3.loc[num, kind] = quadrect(f2_2, num, a[1], b[1], kind)
 
         cls.data2d2 = data3