Changeset View
Changeset View
Standalone View
Standalone View
swh/graph/tests/test_toposort.py
| # Copyright (C) 2022 The Software Heritage developers | # Copyright (C) 2022 The Software Heritage developers | ||||
| # See the AUTHORS file at the top-level directory of this distribution | # See the AUTHORS file at the top-level directory of this distribution | ||||
| # License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| from pathlib import Path | from pathlib import Path | ||||
| import subprocess | import subprocess | ||||
| from swh.graph.luigi import TopoSort | from swh.graph.luigi import TopoSort | ||||
| DATA_DIR = Path(__file__).parents[0] / "dataset" | DATA_DIR = Path(__file__).parents[0] / "dataset" | ||||
| # FIXME: the order of sample ancestors should not be hardcoded | |||||
| # FIXME: swh:1:snp:0000000000000000000000000000000000000022,3,1,swh has three possible | |||||
| # sample ancestors; they should not be hardecoded here | |||||
| EXPECTED = """\ | EXPECTED = """\ | ||||
| SWHID,ancestors,successors,sample_ancestor1,sample_ancestor2 | SWHID,ancestors,successors,sample_ancestor1,sample_ancestor2 | ||||
| swh:1:rev:0000000000000000000000000000000000000003,0,1,, | swh:1:rev:0000000000000000000000000000000000000003,0,1,, | ||||
| swh:1:rev:0000000000000000000000000000000000000009,1,3,swh:1:rev:0000000000000000000000000000000000000003, | swh:1:rev:0000000000000000000000000000000000000009,1,4,swh:1:rev:0000000000000000000000000000000000000003, | ||||
| swh:1:rel:0000000000000000000000000000000000000010,1,1,swh:1:rev:0000000000000000000000000000000000000009, | swh:1:rel:0000000000000000000000000000000000000010,1,2,swh:1:rev:0000000000000000000000000000000000000009, | ||||
| swh:1:snp:0000000000000000000000000000000000000020,2,1,swh:1:rev:0000000000000000000000000000000000000009,swh:1:rel:0000000000000000000000000000000000000010 | swh:1:snp:0000000000000000000000000000000000000020,2,1,swh:1:rev:0000000000000000000000000000000000000009,swh:1:rel:0000000000000000000000000000000000000010 | ||||
| swh:1:ori:83404f995118bd25774f4ac14422a8f175e7a054,1,0,swh:1:snp:0000000000000000000000000000000000000020, | swh:1:ori:83404f995118bd25774f4ac14422a8f175e7a054,1,0,swh:1:snp:0000000000000000000000000000000000000020, | ||||
| swh:1:rev:0000000000000000000000000000000000000013,1,1,swh:1:rev:0000000000000000000000000000000000000009, | swh:1:rev:0000000000000000000000000000000000000013,1,1,swh:1:rev:0000000000000000000000000000000000000009, | ||||
| swh:1:rev:0000000000000000000000000000000000000018,1,1,swh:1:rev:0000000000000000000000000000000000000013, | swh:1:rev:0000000000000000000000000000000000000018,1,2,swh:1:rev:0000000000000000000000000000000000000013, | ||||
| swh:1:rel:0000000000000000000000000000000000000019,1,0,swh:1:rev:0000000000000000000000000000000000000018, | swh:1:rel:0000000000000000000000000000000000000019,1,0,swh:1:rev:0000000000000000000000000000000000000018, | ||||
| swh:1:rel:0000000000000000000000000000000000000021,1,1,swh:1:rev:0000000000000000000000000000000000000018, | |||||
| swh:1:snp:0000000000000000000000000000000000000022,3,1,swh:1:rev:0000000000000000000000000000000000000009,swh:1:rel:0000000000000000000000000000000000000010 | |||||
| swh:1:ori:8f50d3f60eae370ddbf85c86219c55108a350165,1,0,swh:1:snp:0000000000000000000000000000000000000022, | |||||
| """ | """ | ||||
| def test_toposort(tmpdir): | def test_toposort(tmpdir): | ||||
| tmpdir = Path(tmpdir) | tmpdir = Path(tmpdir) | ||||
| topological_order_path = tmpdir / "topo_order.csv.zst" | topological_order_path = tmpdir / "topo_order.csv.zst" | ||||
| Show All 13 Lines | def test_toposort(tmpdir): | ||||
| # The only possible first line | # The only possible first line | ||||
| assert rows[0] == "swh:1:rev:0000000000000000000000000000000000000003,0,1,," | assert rows[0] == "swh:1:rev:0000000000000000000000000000000000000003,0,1,," | ||||
| assert set(rows) == set(expected_lines) | assert set(rows) == set(expected_lines) | ||||
| assert rows.pop() == "", "Missing trailing newline" | assert rows.pop() == "", "Missing trailing newline" | ||||
| # The only two possible last lines | # The only three possible last lines | ||||
| assert rows[-1] in [ | assert rows[-1] in [ | ||||
| "swh:1:ori:83404f995118bd25774f4ac14422a8f175e7a054,1,0" | "swh:1:ori:83404f995118bd25774f4ac14422a8f175e7a054,1,0" | ||||
| ",swh:1:snp:0000000000000000000000000000000000000020,", | ",swh:1:snp:0000000000000000000000000000000000000020,", | ||||
| "swh:1:ori:8f50d3f60eae370ddbf85c86219c55108a350165,1,0" | |||||
| ",swh:1:snp:0000000000000000000000000000000000000022,", | |||||
| "swh:1:rel:0000000000000000000000000000000000000019,1,0" | "swh:1:rel:0000000000000000000000000000000000000019,1,0" | ||||
| ",swh:1:rev:0000000000000000000000000000000000000018,", | ",swh:1:rev:0000000000000000000000000000000000000018,", | ||||
| ] | ] | ||||