def test_average1():
"""
average() nominally computes the average pixel value for a stack of
identically shaped images.
arrays specifies a sequence of inputs arrays, which are nominally a
stack of identically shaped images.
output may be used to specify the output array. If none is specified,
either arrays[0] is copied or a new array of type 'outtype'
is created.
outtype specifies the type of the output array when no 'output' is
specified.
nlow specifies the number of pixels to be excluded from average
on the low end of the pixel stack.
nhigh specifies the number of pixels to be excluded from average
on the high end of the pixel stack.
badmasks specifies boolean arrays corresponding to 'arrays', where true
indicates that a particular pixel is not to be included in the
average calculation.
"""
result = combine.average(arrays)
expected = np.array([[0, 7], [15, 22]])
assert (result == expected).all()
def test_average6():
result = combine.average(arrays,
badmasks=combine.threshhold(arrays, high=25))
expected = np.array([[0, 7], [9, 14]])
assert (result == expected).all()
def test_average5():
bm = np.zeros((4, 2, 2), dtype=np.bool8)
bm[2, ...] = 1
result = combine.average(arrays, badmasks=bm)
expected = np.array([[0, 9], [18, 28]])
assert (result == expected).all()
def test_average4():
result = combine.average(arrays, outtype=np.float32)
expected = np.array([[0., 7.5], [15., 22.5]], dtype=np.float32)
assert (result == expected).all()
def test_average3():
result = combine.average(arrays, nlow=1)
expected = np.array([[0, 9], [18, 28]])
assert (result == expected).all()
def test_average2():
result = combine.average(arrays, nhigh=1)
expected = np.array([[0, 4], [9, 14]])
assert (result == expected).all()