diff options
| author | Arun Isaac | 2026-08-14 21:46:33 +0100 |
|---|---|---|
| committer | Arun Isaac | 2026-08-15 20:53:15 +0100 |
| commit | 72f63f7817579d7dd678f771f0a0b65111543f67 (patch) | |
| tree | b201bf4d49c44b7e26ed0e034765dda13ebc115a /tests | |
| parent | ce5bc77c767351c57ce4f7872c827316cb3b1040 (diff) | |
| download | domagi-72f63f7817579d7dd678f771f0a0b65111543f67.tar.gz domagi-72f63f7817579d7dd678f771f0a0b65111543f67.tar.lz domagi-72f63f7817579d7dd678f771f0a0b65111543f67.zip | |
When sorting a data frame by a single column, only pass in a string.
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_domagi.py | 10 |
1 files changed, 5 insertions, 5 deletions
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) |