# Path: impyute/deletion/complete_case.py
# @wrapper.wrappers
# @wrapper.checks
# def complete_case(data):
#     """ Return only data rows with all columns
# 
#     Parameters
#     ----------
#     data: numpy.ndarray
#         Data to impute.
# 
#     Returns
#     -------
#     numpy.ndarray
#         Imputed data.
# 
#     """
#     return data[~np.isnan(data).any(axis=1)]
# 
# 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 numpy as np
from impyute.deletion import complete_case
from impyute.ops.testing import return_na_check

SHAPE = (5, 5)


def test_complete_case_(test_data):
    data = test_data(SHAPE)
    imputed = complete_case(data)
    return_na_check(imputed)


def test_impute_missing_values(test_data):
    data = test_data(SHAPE)
    imputed = complete_case(data)
    assert np.shape(imputed) == (4, 5)


def test_imputed_values(test_data):
    data = test_data(SHAPE)

# Path: impyute/deletion/complete_case.py
# @wrapper.wrappers
# @wrapper.checks
# def complete_case(data):
#     """ Return only data rows with all columns
# 
#     Parameters
#     ----------
#     data: numpy.ndarray
#         Data to impute.
# 
#     Returns
#     -------
#     numpy.ndarray
#         Imputed data.
# 
#     """
#     return data[~np.isnan(data).any(axis=1)]
# 
# 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/deletion/test_complete_case.py
import numpy as np
from impyute.deletion import complete_case
from impyute.ops.testing import return_na_check


SHAPE = (5, 5)


def test_complete_case_(test_data):
    data = test_data(SHAPE)
    imputed = complete_case(data)
    return_na_check(imputed)


def test_impute_missing_values(test_data):
    data = test_data(SHAPE)
    imputed = complete_case(data)
    assert np.shape(imputed) == (4, 5)


def test_imputed_values(test_data):
    data = test_data(SHAPE)

    imputed = complete_case(data)