110 lines
3.7 KiB
Python
110 lines
3.7 KiB
Python
import io
|
|
import pytest
|
|
import numpy
|
|
from datetime import datetime
|
|
from klamath.basic import KlamathError
|
|
from klamath import records
|
|
|
|
def test_record_tags() -> None:
|
|
assert records.HEADER.tag == 0x0002
|
|
assert records.BGNLIB.tag == 0x0102
|
|
assert records.LIBNAME.tag == 0x0206
|
|
assert records.UNITS.tag == 0x0305
|
|
assert records.ENDLIB.tag == 0x0400
|
|
assert records.BGNSTR.tag == 0x0502
|
|
assert records.STRNAME.tag == 0x0606
|
|
assert records.ENDSTR.tag == 0x0700
|
|
assert records.BOUNDARY.tag == 0x0800
|
|
assert records.PATH.tag == 0x0900
|
|
assert records.SREF.tag == 0x0a00
|
|
assert records.AREF.tag == 0x0b00
|
|
assert records.TEXT.tag == 0x0c00
|
|
assert records.LAYER.tag == 0x0d02
|
|
assert records.DATATYPE.tag == 0x0e02
|
|
assert records.WIDTH.tag == 0x0f03
|
|
assert records.XY.tag == 0x1003
|
|
assert records.ENDEL.tag == 0x1100
|
|
assert records.SNAME.tag == 0x1206
|
|
assert records.COLROW.tag == 0x1302
|
|
assert records.NODE.tag == 0x1500
|
|
assert records.TEXTTYPE.tag == 0x1602
|
|
assert records.PRESENTATION.tag == 0x1701
|
|
assert records.STRING.tag == 0x1906
|
|
assert records.STRANS.tag == 0x1a01
|
|
assert records.MAG.tag == 0x1b05
|
|
assert records.ANGLE.tag == 0x1c05
|
|
assert records.REFLIBS.tag == 0x1f06
|
|
assert records.FONTS.tag == 0x2006
|
|
assert records.PATHTYPE.tag == 0x2102
|
|
assert records.GENERATIONS.tag == 0x2202
|
|
assert records.ATTRTABLE.tag == 0x2306
|
|
assert records.ELFLAGS.tag == 0x2601
|
|
assert records.NODETYPE.tag == 0x2a02
|
|
assert records.PROPATTR.tag == 0x2b02
|
|
assert records.PROPVALUE.tag == 0x2c06
|
|
assert records.BOX.tag == 0x2d00
|
|
assert records.BOXTYPE.tag == 0x2e02
|
|
assert records.PLEX.tag == 0x2f03
|
|
assert records.BGNEXTN.tag == 0x3003
|
|
assert records.ENDEXTN.tag == 0x3103
|
|
assert records.TAPENUM.tag == 0x3202
|
|
assert records.TAPECODE.tag == 0x3302
|
|
assert records.FORMAT.tag == 0x3602
|
|
assert records.MASK.tag == 0x3706
|
|
assert records.ENDMASKS.tag == 0x3800
|
|
assert records.LIBDIRSIZE.tag == 0x3902
|
|
assert records.SRFNAME.tag == 0x3a06
|
|
assert records.LIBSECUR.tag == 0x3b02
|
|
|
|
def test_header_validation() -> None:
|
|
# Correct size
|
|
records.HEADER.check_size(2)
|
|
|
|
# Incorrect size
|
|
with pytest.raises(KlamathError, match="Expected size 2, got 4"):
|
|
records.HEADER.check_size(4)
|
|
|
|
def test_bgnlib_validation() -> None:
|
|
now = datetime(2023, 10, 27, 12, 30, 45)
|
|
# Correct size (2 datetimes = 24 bytes)
|
|
records.BGNLIB.check_size(24)
|
|
|
|
# Incorrect size
|
|
with pytest.raises(KlamathError, match="Expected size 24, got 12"):
|
|
records.BGNLIB.check_size(12)
|
|
|
|
def test_reflibs_fonts_validation() -> None:
|
|
# REFLIBS must be multiple of 44
|
|
records.REFLIBS.check_size(44)
|
|
records.REFLIBS.check_size(88)
|
|
records.REFLIBS.check_size(0)
|
|
|
|
with pytest.raises(KlamathError, match="Expected size to be multiple of 44"):
|
|
records.REFLIBS.check_size(10)
|
|
|
|
def test_generations_format_validation() -> None:
|
|
# GENERATIONS expects exactly one integer
|
|
records.GENERATIONS.check_data(3)
|
|
records.GENERATIONS.check_data([1])
|
|
|
|
with pytest.raises(KlamathError, match="Expected exactly one integer"):
|
|
records.GENERATIONS.check_data([1, 2])
|
|
|
|
def test_attrtable_validation() -> None:
|
|
# ATTRTABLE size <= 44
|
|
records.ATTRTABLE.check_size(44)
|
|
records.ATTRTABLE.check_size(10)
|
|
|
|
with pytest.raises(KlamathError, match="Expected size <= 44"):
|
|
records.ATTRTABLE.check_size(45)
|
|
|
|
def test_nodata_records() -> None:
|
|
stream = io.BytesIO()
|
|
records.ENDLIB.write(stream, None)
|
|
stream.seek(0)
|
|
assert records.ENDLIB.read(stream) is None
|
|
|
|
stream = io.BytesIO()
|
|
records.BOUNDARY.write(stream, None)
|
|
stream.seek(0)
|
|
assert records.BOUNDARY.read(stream) is None
|