test_illumination_statistics.py 文件源码

def test_add_to_mean_with_scaling(illumination_stats):
    numpy.random.seed(8)
    image = numpy.random.randint(256, size=(16, 16, 3), dtype=numpy.uint16)

    illumination_stats.addToMean(image)

    assert illumination_stats.mean_image.shape == (8, 8, 3)
    result_mean = illumination_stats.mean_image
    assert result_mean.sum() > 0
    #numpy.testing.assert_array_equal(result_mean.astype(numpy.uint16), image)