AnasSarkiz/NEMA14_CH32X035G8U6

tests/test_placement_advisories.py

import pathlib
import re
import sys
import unittest

ROOT = pathlib.Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT / 'scripts'))
from placement_advisories import orientation_advisories_only


class PlacementAdvisoryGateTests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.log = (ROOT / 'artifacts/validation-resume/native_placement.txt').read_text()

    def test_current_known_advisories_remain_reviewable(self):
        self.assertTrue(orientation_advisories_only(self.log))

    def test_summary_cannot_hide_real_drc_error(self):
        self.assertFalse(orientation_advisories_only(self.log.replace('Errors: 0', 'Errors: 1')))

    def test_summary_cannot_hide_drc_warning(self):
        self.assertFalse(orientation_advisories_only(self.log.replace('Warnings: 0', 'Warnings: 1')))

    def test_unrecognized_issue_requires_review(self):
        self.assertFalse(orientation_advisories_only(self.log.replace(re.search(r'Suggested move: rotate [A-Za-z0-9_]+ 180 degrees\.', self.log).group(0), 'Unexpected placement defect.')))

    def test_trailing_diagnostics_are_not_ignored(self):
        self.assertFalse(orientation_advisories_only(self.log + 'Unexpected diagnostic\n'))

    def test_truncated_worst_list_requires_complete_flagged_components(self):
        ten = (ROOT / 'tests/fixtures/placement-ten-orientations.txt').read_text()
        self.assertTrue(orientation_advisories_only(ten))
        self.assertFalse(orientation_advisories_only(ten.replace('10 suboptimal orientations', '11 suboptimal orientations')))

    def test_unknown_flagged_issue_cannot_hide_after_worst_list(self):
        ten = (ROOT / 'tests/fixtures/placement-ten-orientations.txt').read_text()
        prefix, flagged = ten.split('flagged components:', 1)
        self.assertFalse(orientation_advisories_only(prefix + 'flagged components:' + flagged.replace('Suggested move: rotate C_PD 180 degrees.', 'Unexpected placement defect.', 1)))

    def test_missing_drc_section_requires_review(self):
        self.assertFalse(orientation_advisories_only(self.log.split('placement drc:')[0]))


if __name__ == '__main__':
    unittest.main()