# 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: impyute/ops/error.py
# class BadInputError(Exception):
# class BadOutputError(Exception):


import numpy as np
import impyute as impy
from impyute.ops.testing import return_na_check
from impyute.ops import error
"""test_locf.py"""

SHAPE = (5, 5)


def test_locf_(test_data):
    data = test_data(SHAPE)
    imputed = impy.locf(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: impyute/ops/error.py
# class BadInputError(Exception):
# class BadOutputError(Exception):


# Path: test/imputation/ts/test_locf.py
import numpy as np
import impyute as impy
from impyute.ops.testing import return_na_check
from impyute.ops import error

"""test_locf.py"""

SHAPE = (5, 5)


def test_locf_(test_data):
    data = test_data(SHAPE)
    imputed = impy.locf(data)

    return_na_check(imputed)