.

pygridgen · Oct 25, 2024 · f666048 · f666048
1 parent 42de813
commit f666048
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pygridtools/tests/test_iotools.py b/pygridtools/tests/test_iotools.py
@@ -76,10 +76,10 @@ def test_read_grid():
     known_df = pandas.DataFrame(
         data={'easting': [1., 2., 3.] * 4, 'northing': sorted([4., 5., 6., 7.] * 3)},
         index=pandas.MultiIndex.from_product([[2, 3, 4, 5], [2, 3, 4]], names=['jj', 'ii'])
-    ).assign(elev=0.0).assign(river='test').reset_index().set_index(['ii', 'jj']).sort_index().assign(
+    ).assign(elev=0.0).assign(river='test').reset_index().assign(
         ii=lambda df: df["ii"].astype("int64"),
         jj=lambda df: df["jj"].astype("int64"),
-    )
+    ).set_index(['ii', 'jj']).sort_index()
 
     pdtest.assert_frame_equal(result_df, known_df)