test_outputport.py

"""
B-ASIC test suite for OutputPort
TODO: More info
"""
from b_asic import InputPort, OutputPort
import pytest

@pytest.fixture
def inp_ports():
    return [InputPort(_, None) for _ in range(0,3)]

def test_connect_multiple_signals(inp_ports):
    """Can multiple ports connect to an output port?"""
    out_port = OutputPort(0, None)

    for port in inp_ports:
        out_port.connect_port(port)
    
    assert out_port.signal_count() == len(inp_ports)

def test_disconnect_multiple_signals(inp_ports):
    """Can multiple ports disconnect from an output port?"""
    out_port = OutputPort(0, None)

    for port in inp_ports:
        out_port.connect_port(port)
    
    for _ in inp_ports:
        out_port.disconnect_signal(0)

    assert out_port.signal_count() == 0