1# Copyright (C) 2016-2020, Manfred Moitzi 2# License: MIT License 3import pytest 4import codecs 5from ezdxf.lldxf.encoding import dxf_backslash_replace, encode 6from pathlib import Path 7 8# setup DXF unicode encoder -> '\U+nnnn' 9codecs.register_error('dxfreplace', dxf_backslash_replace) 10 11 12def test_ascii_encoding(): 13 assert b'123\\U+6539' == encode('123改', 'ascii') 14 15 16@pytest.mark.parametrize(['s', 'e'], [ 17 ('300\n\udcb7\udc9e\udcff\n', b'300\n\xb7\x9e\xff\n'), 18 ('123改', b'123\\U+6539') 19]) 20def test_surrogate_escape_support_in_dxf_replace_encoder(s, e): 21 assert e == encode(s, 'ascii') 22 23 24@pytest.mark.parametrize('n', [0, 1, 2]) 25def test_XRECORD_handling_of_dxf_replace_encoder(n): 26 XRECORD = Path(__file__).parent / f'XRECORD_{n}.bin' 27 with open(XRECORD, 'rb') as f: 28 data = f.read() 29 s = data.decode('utf8', errors='surrogateescape') 30 result = encode(s, encoding='utf8') 31 assert data == result 32