import numpy as np
import python_example as m
import pytest


def test_version():
    assert m.__version__ == "0.0.1"


def test_add_1_2():
    assert m.add(1, 2) == 3


def test_subtract_1_2():
    assert m.subtract(1, 2) == -1


def test_ndarray_args():
    data = np.array([range(10)])
    assert m.ndarray_sum(data) == sum(range(10))


def test_runtime_error():
    with pytest.raises(RuntimeError):
        m.runtime_error()


def test_range_error():
    with pytest.raises(ValueError):
        m.range_error()


def test_return_ndarray():
    a = m.get_ndarray(10, 5.0)
    assert isinstance(a, np.ndarray)
    np.testing.assert_allclose(a, 5.0)
    assert len(a) == 10


def test_get_class_with_vector_results():
    n = 10
    t = 4.3
    h = 88.3
    results = m.get_vector_results(n, t, h)
    assert len(results.temperature) == n
    np.testing.assert_allclose(results.temperature, t)
    assert len(results.humidity) == n
    np.testing.assert_allclose(results.humidity, h)