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