from codonPython.check_null import check_null

import numpy as np

import pandas as pd

import pytest

testdata = pd.DataFrame({

"col1" : [1,2,3,4,5,6,7,8,9,10],

"col2" : [11,12,13,14,15,np.nan,np.nan,18,19,20],

})

@pytest.mark.parametrize("dataframe, columns_to_be_checked, expected", [

(testdata.iloc[:5, :], ["col1", "col2"], 0),

(testdata, ["col2"], 2),

])

def test_BAU(dataframe, columns_to_be_checked, expected):

assert check_null(dataframe, columns_to_be_checked) == expected

@pytest.mark.parametrize("dataframe, columns_to_be_checked", [

(testdata, 0.01),

])

def test_ValueError(dataframe, columns_to_be_checked):

with pytest.raises(ValueError):

check_null(dataframe, columns_to_be_checked)

@pytest.mark.parametrize("dataframe, columns_to_be_checked", [

(testdata, ["wrong_column"]),

])

def test_KeyError(dataframe, columns_to_be_checked):

with pytest.raises(KeyError):

check_null(dataframe, columns_to_be_checked)