signal.py

import pytest
from b_asic import Signal, SignalSource, SignalDestination, Addition

"""
Use a fixture for initializing objects and pass them as argument to a test function
"""
@pytest.fixture
def signal():
    source = SignalSource(Addition(), 1)
    dest = SignalDestination(Addition(), 2)
    return Signal(source, dest)

@pytest.fixture
def signals():
    ret = []
    for _ in range(0,3):
        source = SignalSource(Addition(), 1)
        dest = SignalDestination(Addition(), 2)
        ret.append(Signal(source, dest))
    return ret