"""Tests for kindlepid PID/serial helpers."""

import dedrm_test_utils as U

kp = U.load("kindlepid", package="dedrm")


def test_checksum_pid_appends_two_chars():
    pid = kp.checksumPid(b"12345678")
    assert pid == b"12345678EL"
    assert len(pid) == 10


def test_pid_from_serial_known_vector():
    assert kp.pidFromSerial(b"B00212345678", 8) == b"VBKTRX7Q"