def test_get_soil_group_fail(mocker): # Mock the variables class variables = mocker.Mock() test = Loads(variables) with pytest.raises(ValueError): test._get_soil_group("bob")
def test_get_soil_group(mocker, soiltype, expected): # Mock the variables class variables = mocker.Mock() test = Loads(variables) result = test._get_soil_group(soiltype) assert result == expected