import pytest

from lab_control.measurement import *

@pytest.fixture
def uut():
    return Measurement()

def test_linearRangeAscending():
    first = 100.0
    last = 200.0
    steps = 11

    expectedRange = [
        100.0, 110.0, 120.0, 130.0,
        140.0, 150.0, 160.0, 170.0,
        180.0, 190.0, 200.0
    ]

    actualRange = getLinearRange(first, last, steps)
    assert actualRange == expectedRange

def test_linearRangeDescending():
    first = 200.0
    last = 180.0
    steps = 3

    expectedRange = [
        200.0, 190.0, 180.0
    ]

    actualRange = getLinearRange(first, last, steps)
    assert actualRange == expectedRange

def test_linearRangeTwoSteps():
    first = 20.0
    last = -20.0
    steps = 2

    expectedRange = [
        20.0, -20.0
    ]

    actualRange = getLinearRange(first, last, steps)
    assert actualRange == expectedRange

def test_linearRangeOneStep():
    first = 20.0
    last = -20.0
    steps = 1

    with pytest.raises(AssertionError):
        actualRange = getLinearRange(first, last, steps)