65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
# SPDX-FileCopyrightText: KB Sriram
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
"""
|
|
Tests in
|
|
"""
|
|
|
|
import pytest
|
|
from pytest_helpers import assert_assembles_to, assert_assembly_fails
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"source,expected",
|
|
[
|
|
("pins", 0b000),
|
|
("x", 0b001),
|
|
("y", 0b010),
|
|
("null", 0b011),
|
|
("isr", 0b110),
|
|
("osr", 0b111),
|
|
],
|
|
)
|
|
def test_in_sources(source: str, expected: int) -> None:
|
|
# delay src bitcount
|
|
encoding = 0b010_00000_000_10001
|
|
# add in the expected source
|
|
encoding |= expected << 5
|
|
assert_assembles_to(f"in {source}, 17", [encoding])
|
|
|
|
|
|
@pytest.mark.parametrize("delay", [0, 1, 9, 17, 31])
|
|
def test_in_delay(delay: int) -> None:
|
|
# delay src bitcount
|
|
encoding = 0b010_00000_000_10001
|
|
# add in the expected delay
|
|
encoding |= delay << 8
|
|
assert_assembles_to(f"in pins, 17 [{delay}]", [encoding])
|
|
|
|
|
|
@pytest.mark.parametrize("bitcount", [1, 9, 17, 32])
|
|
def test_in_bitcount(bitcount: int) -> None:
|
|
# delay dst bitcount
|
|
encoding = 0b010_00000_000_00000
|
|
# add in the expected bitcount. Note that
|
|
# 32 should be encoded as 0, which we do by
|
|
# masking the bitcount with 0x1f
|
|
encoding |= bitcount & 0x1F
|
|
assert_assembles_to(f"in pins, {bitcount}", [encoding])
|
|
|
|
|
|
def test_in_delay_with_sideset() -> None:
|
|
source = [
|
|
".side_set 2",
|
|
"in pins 17 side 2 [5]",
|
|
]
|
|
assert_assembles_to("\n".join(source), [0b010_10_101_000_10001])
|
|
|
|
|
|
def test_in_bad_source() -> None:
|
|
assert_assembly_fails("in bad, 17", match="Invalid in source 'bad'", errtype=ValueError)
|
|
|
|
|
|
def test_in_bad_bitcount() -> None:
|
|
assert_assembly_fails("in pins, 0", match="Count out of range", errtype=RuntimeError)
|