Something went wrong. Try again.
Personal fork of prime agent. Mirror of: https://mnemosyne.sigilzero.dev/sigilzero/prime-agent
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566#!/usr/bin/env python3"""Lightweight capture parser checks; never execute product binaries."""import unittestfrom review_markdown_frames_parity import StyledScreen, START, END, CORPUS
class CaptureTests(unittest.TestCase): def test_captured_keyboard_controls_are_not_sgr(self): screen = StyledScreen(40) screen.feed(b'\x1b[1mA\x1b[>4;0mB') replies = screen.feed(b'\x1b[?uC') self.assertEqual(replies, []) self.assertEqual(screen.keyboard_controls, ['\x1b[>4;0m', '\x1b[?u']) self.assertEqual(screen.styles[0][:3], [(None, None, (1,), None)] * 3) self.assertEqual(screen.style_errors, [])
def test_partial_keyboard_query_and_config_wait_for_terminator(self): screen = StyledScreen(40) self.assertEqual(screen.feed(b'\x1b[>4;'), []) self.assertEqual(screen.keyboard_controls, []) screen.feed(b'0m\x1b[?') self.assertEqual(screen.keyboard_controls, ['\x1b[>4;0m']) self.assertEqual(screen.feed(b'u\x1b[c'), [b'\x1b[?1;2c']) self.assertEqual(screen.style_errors, [])
def test_unknown_private_styles_and_geometry_still_fail(self): screen = StyledScreen(40) screen.feed(b'\x1b[>31m\x1b[3S') self.assertEqual(len(screen.style_errors), 2) self.assertEqual(screen.fg, None) self.assertEqual(screen.mods, set())
def test_fragmented_sgr_and_osc8_keep_cell_attributes(self): screen = StyledScreen(40) chunks = [b'\x1b[38;2;1;', b'2;3m\x1b[1mA\x1b[22mB', b'\x1b]8;;https://example.com\x1b\\C\x1b]8;;\x1b\\', b'D'] for chunk in chunks: screen.feed(chunk) self.assertEqual(screen.cells[0][:4], list('ABCD')) self.assertEqual(screen.styles[0][:4], [ (('rgb', 1, 2, 3), None, (1,), None), (('rgb', 1, 2, 3), None, (), None), (('rgb', 1, 2, 3), None, (), 'https://example.com'), (('rgb', 1, 2, 3), None, (), None)]) self.assertEqual(screen.style_errors, [])
def test_repaint_clears_stale_text_and_style(self): screen = StyledScreen(40) screen.feed(b'\x1b[1mABCDE\r\x1b[0mXY\x1b[K') self.assertEqual(screen.cells[0][:5], list('XY ')) self.assertEqual(screen.styles[0][2:5], [None] * 3)
def test_sentinel_crop_preserves_indentation_and_nonblank_styles(self): screen = StyledScreen(40) screen.feed(f' {START}\r\n '.encode() + b'\x1b[3mtext\x1b[0m\r\n' + f' {END}'.encode()) self.assertEqual(screen.region(), [[(' ', None), (' ', None)] + [(char, (None, None, (3,), None)) for char in 'text']])
def test_title_corpus_has_single_backslash_escapes(self): self.assertIn(r'\"b\"', CORPUS['escaped_titles']) self.assertNotIn(r'\\"', CORPUS['escaped_titles'])
if __name__ == '__main__': unittest.main()