fix unit tests

Author: dougbrn

src/nested_dask/core.py

@@ -647,8 +647,10 @@ def sort_values(
             The ideal number of output partitions. If None, use the same as the
             input. If ‘auto’ then decide by memory use. Not used when sorting
             nested layers.
-        ascending: bool, optional
-            Sort ascending vs. descending. Defaults to True.
+        ascending: bool or list[bool], optional
+            Sort ascending vs. descending. Defaults to True. Specify list for
+            multiple sort orders. If this is a list of bools, must match the
+            length of the by.
         na_position: {‘last’, ‘first’}, optional
             Puts NaNs at the beginning if ‘first’, puts NaN at the end if
             ‘last’. Defaults to ‘last’.

tests/nested_dask/test_nestedframe.py

@@ -308,8 +308,8 @@ def test_sort_values(test_dataset):
 
     # test sorting on nested columns
     sorted_nested = test_dataset.sort_values(by="nested.flux", ascending=False)
-    assert sorted_nested.compute().loc[0]["nested"]["flux"].values.tolist() == sorted(
-        test_dataset.loc[0]["nested.flux"].values.compute().tolist(),
+    assert sorted_nested.compute().iloc[0]["nested"]["flux"].values.tolist() == sorted(
+        test_dataset.compute().iloc[0]["nested"]["flux"].values.tolist(),
         reverse=True,
     )
     assert sorted_nested.known_divisions  # Divisions should be known