From 2da720dc7700f9ffc22b0d7e5351090f6766362e Mon Sep 17 00:00:00 2001 From: njzjz-bot Date: Fri, 17 Jul 2026 07:35:14 +0800 Subject: [PATCH] test(dpmodel): collect ASE virtual atom regression Move the existing ASE virtual-atom parity test into TestAseCarryAll so pytest collects it. This locks in the runtime filtering added by PR #5714. Coding-Agent: Codex Codex-Version: codex-cli 0.144.4 Model: gpt-5.6-sol Reasoning-Effort: xhigh --- source/tests/common/dpmodel/test_from_ijs.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/source/tests/common/dpmodel/test_from_ijs.py b/source/tests/common/dpmodel/test_from_ijs.py index dc0af94037..adce99c5e5 100644 --- a/source/tests/common/dpmodel/test_from_ijs.py +++ b/source/tests/common/dpmodel/test_from_ijs.py @@ -85,10 +85,6 @@ def test_ase_matches_intree_carry_all_nonperiodic(self) -> None: ng_ref = build_neighbor_graph(coord, atype, None, rcut=4.0) self.assertEqual(self._sets(ng_ase, 6), self._sets(ng_ref, 6)) - -if __name__ == "__main__": - unittest.main() - def test_ase_excludes_virtual_atoms_like_dense(self) -> None: """Virtual atoms (atype < 0) excluded as center AND neighbor. @@ -113,3 +109,7 @@ def test_ase_excludes_virtual_atoms_like_dense(self) -> None: ei = ng_ase.edge_index[:, ng_ase.edge_mask] flat_atype = atype.reshape(-1) assert np.all(flat_atype[ei[0]] >= 0) and np.all(flat_atype[ei[1]] >= 0) + + +if __name__ == "__main__": + unittest.main()