# Path: impyute/ops/testing.py
# def return_na_check(data):
#     """Helper function for tests to check if the data returned is a
#        numpy array and that the imputed data has no NaN's.
# 
#     Parameters
#     ----------
#     data: numpy.ndarray
#         Data to impute.
#     
#     Returns
#     -------
#     None
# 
#     """
#     assert isinstance(data, np.ndarray)
#     assert not np.isnan(data).any()


import impyute as impy
from impyute.ops.testing import return_na_check
"""test_averagings.py"""

SHAPE = (5, 5)


def test_mean(test_data):
    data = test_data(SHAPE)
    imputed = impy.mean(data)
    return_na_check(imputed)


def test_mode(test_data):
    data = test_data(SHAPE)
    imputed = impy.mode(data)

# Path: impyute/ops/testing.py
# def return_na_check(data):
#     """Helper function for tests to check if the data returned is a
#        numpy array and that the imputed data has no NaN's.
# 
#     Parameters
#     ----------
#     data: numpy.ndarray
#         Data to impute.
#     
#     Returns
#     -------
#     None
# 
#     """
#     assert isinstance(data, np.ndarray)
#     assert not np.isnan(data).any()


# Path: test/imputation/cs/test_central_tendency.py
import impyute as impy
from impyute.ops.testing import return_na_check

"""test_averagings.py"""

SHAPE = (5, 5)


def test_mean(test_data):
    data = test_data(SHAPE)
    imputed = impy.mean(data)
    return_na_check(imputed)


def test_mode(test_data):
    data = test_data(SHAPE)
    imputed = impy.mode(data)

    return_na_check(imputed)