From 72f63f7817579d7dd678f771f0a0b65111543f67 Mon Sep 17 00:00:00 2001
From: Arun Isaac
Date: Fri, 14 Aug 2026 21:46:33 +0100
Subject: When sorting a data frame by a single column, only pass in a string.

---
 tests/test_domagi.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'tests')

diff --git a/tests/test_domagi.py b/tests/test_domagi.py
index 1ce5fc0..d678a3f 100644
--- a/tests/test_domagi.py
+++ b/tests/test_domagi.py
@@ -74,7 +74,7 @@ def test_domagi_crush(tmp_path, test_data_file, expected_output):
                            Path("test-data/expected-output/test3-depth"))])
 def test_domagi_depth(tmp_path, test_data_file, expected_output):
     expected = (pd.read_csv(expected_output, sep="\t")
-                .sort_values(by=["#path"], ignore_index=True))
+                .sort_values(by="#path", ignore_index=True))
     duckdb_path = tmp_path / f"{test_data_file.stem}.db"
     runner = CliRunner()
     result = runner.invoke(main, ["build",
@@ -87,7 +87,7 @@ def test_domagi_depth(tmp_path, test_data_file, expected_output):
     assert_frame_equal(expected,
                        pd.read_csv(io.StringIO(result.stdout),
                                    sep="\t")
-                       .sort_values(by=["#path"],
+                       .sort_values(by="#path",
                                     ignore_index=True),
                        check_dtype=False)
     for _, row in expected.iterrows():
@@ -100,7 +100,7 @@ def test_domagi_depth(tmp_path, test_data_file, expected_output):
         assert_frame_equal(per_path_expected,
                            pd.read_csv(io.StringIO(result.stdout),
                                        sep="\t")
-                           .sort_values(by=["#path"],
+                           .sort_values(by="#path",
                                         ignore_index=True),
                            check_dtype=False)
 
@@ -123,11 +123,11 @@ def test_domagi_depth_graph_depth(tmp_path, test_data_file, expected_output):
                                   "--db", duckdb_path])
     assert result.exit_code == 0
     assert_frame_equal(pd.read_csv(expected_output, sep="\t")
-                       .sort_values(by=["#node.id"],
+                       .sort_values(by="#node.id",
                                     ignore_index=True),
                        pd.read_csv(io.StringIO(result.stdout),
                                    sep="\t")
-                       .sort_values(by=["#node.id"],
+                       .sort_values(by="#node.id",
                                     ignore_index=True),
                        check_dtype=False)
 
-- 
cgit 1.4.1