def test_3D_axis1():
open(logfile, 'w').close()
checks = {'MEAN_IN_RANGE': (0, 7)}
arr = arr_3D.copy()
arr[np.isnan(arr)] = 0
arr[arr < 0] = 0
ck.check(arr, checks, data_tag, logger, 1)
assert is_text_in_file(logfile, 'frame #0')
assert is_text_in_file(logfile, 'frame #1')
assert not is_text_in_file(logfile, 'frame #2')
def test_no_sat_no_in_range():
checks = {'SAT_IN_RANGE': (1, 2)}
arr = arr_3D.copy()
arr[np.isnan(arr)] = 0
arr[arr < 0] = 0
verified = ck.check(arr, checks, data_tag, logger)
assert not verified
def test_is_mean_in_range():
checks = {'MEAN_IN_RANGE': (0, 7)}
arr = arr_2D.copy()
arr[np.isnan(arr)] = 0
arr[arr < 0] = 0
verified = ck.check(arr, checks, data_tag, logger)
assert verified
def test_no_mean_no_in_range():
checks = {'MEAN_IN_RANGE': (100, 107)}
arr = arr_3D.copy()
arr[np.isnan(arr)] = 0
arr[arr < 0] = 0
verified = ck.check(arr, checks, data_tag, logger)
assert not verified
def test_is_complex():
checks = {'IS_COMPLEX': ()}
arr = arr_3D.copy()
arr[np.isnan(arr)] = 0
arr = arr.astype(np.dtype(np.complex))
verified = ck.check(arr, checks, data_tag, logger)
assert verified
def test_no_float():
checks = {'IS_FLOAT': ()}
arr = arr_3D.copy()
arr[np.isnan(arr)] = 0
arr = arr.astype(int)
verified = ck.check(arr, checks, data_tag, logger)
assert not verified
def test_is_int():
checks = {'IS_INT': ()}
arr = arr_3D.copy()
arr[np.isnan(arr)] = 0
arr = arr.astype(int)
verified = ck.check(arr, checks, data_tag, logger)
assert verified
def test_no_tag():
open('default.log', 'w').close()
checks = {'IS_NPARRAY': ()}
verified = ck.check(arr_2D, checks)
assert verified
assert is_text_in_file(
'default.log', 'mydata' + ' evaluated "is_nparray" with result True')
os.remove('default.log')
def test_3D_no_axis():
open(logfile, 'w').close()
checks = {'MEAN_IN_RANGE': (0, 7)}
arr = arr_2D.copy()
arr[np.isnan(arr)] = 0
arr[arr < 0] = 0
verified = ck.check(arr, checks, data_tag, logger)
assert verified
assert is_text_in_file(
logfile,
data_tag + ' evaluated frame #0 mean_in_range with result True')
assert not is_text_in_file(logfile, 'frame #1')
def test_no_size_dims():
checks = {'IS_SIZE': arr_3D.shape}
verified = ck.check(arr_2D, checks, data_tag, logger)
assert not verified
def test_no_logger():
checks = {'IS_NPARRAY': ()}
verified = ck.check(arr_2D, checks, data_tag)
assert verified
assert is_text_in_file(
'default.log', data_tag + ' evaluated "is_nparray" with result True')
def test_no_nans():
checks = {'HAS_NO_NAN': ()}
arr = arr_3D.copy()
arr[np.isnan(arr)] = 0
verified = ck.check(arr, checks, data_tag, logger)
assert verified
def test_no_negative():
checks = {'HAS_NO_NEGATIVE': ()}
arr = arr_2D.copy()
arr[arr < 0] = 0
verified = ck.check(arr, checks, data_tag, logger)
assert verified
def test_nparray():
checks = {'IS_NPARRAY': ()}
verified = ck.check(arr_2D, checks, data_tag, logger)
assert verified
def test_is_size():
checks = {'IS_SIZE': (4, 2, 3)}
verified = ck.check(arr_3D, checks, data_tag, logger)
assert verified
def test_no_complex():
checks = {'IS_COMPLEX': ()}
arr = arr_3D.copy()
arr[np.isnan(arr)] = 0
verified = ck.check(arr, checks, data_tag, logger)
assert not verified
def test_has_nans():
checks = {'HAS_NO_NAN': ()}
verified = ck.check(arr_3D, checks, data_tag, logger)
assert not verified
def test_no_nparra():
checks = {'IS_NPARRAY': ()}
verified = ck.check('a', checks, data_tag, logger)
assert not verified
def test_no_size_shape():
checks = {'IS_SIZE': (9, 10)}
verified = ck.check(arr_2D, checks, data_tag, logger)
assert not verified
def test_has_negative():
checks = {'HAS_NO_NEGATIVE': ()}
verified = ck.check(arr_3D, checks, data_tag, logger)
assert not verified
import censor.checks as ck
import numpy as np
arr_3D = np.array([[[1, 2, 3], [np.log(-1.), -5, -7]],[[1, 2, 3], [4, 5, 6]],
[[1, 2, 3], [4, 5, 6]], [[1, 2, 3], [4, 5, 6]]])
checks = {'MEAN_IN_RANGE': (0, 7)}
arr = arr_3D.copy()
arr[np.isnan(arr)] = 0
arr[arr < 0] = 0
ck.check(arr, checks)
