from codonPython.nhsNumber import nhsNumberGenerator, nhsNumberValidator

import numpy as np

import pytest

import random

@pytest.mark.parametrize("to_generate, random_state, expected", [

(3, 42, [8429141456, 2625792787, 8235363119]),

(2, 1, [9598980006, 6597925149])

])

def test_nhsNumberGenerator_BAU(to_generate, random_state, expected):

assert expected == nhsNumberGenerator(

to_generate, random_state=random_state)

@pytest.mark.parametrize("to_generate", [

4.2,

1000001,

-1

])

def test_nhsNumberGenerator_valueErrors(to_generate):

with pytest.raises(ValueError):

nhsNumberGenerator(to_generate)

@pytest.mark.parametrize("to_validate, expected", [

(9598980006, True),

(9598980007, False)

])

def test_nhsNumberValidator_BAU(to_validate, expected):

assert expected == nhsNumberValidator(to_validate)

@pytest.mark.parametrize("to_validate", [

4.2,

1000001,

-1

])

def test_nhsNumberValidator_valueErrors(to_validate):

with pytest.raises(ValueError):

nhsNumberValidator(to_validate)