# 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)