testOpFilterLabels.py

from builtins import object
###############################################################################
#   lazyflow: data flow based lazy parallel computation framework
#
#       Copyright (C) 2011-2014, the ilastik developers
#                                <team@ilastik.org>
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the Lesser GNU General Public License
# as published by the Free Software Foundation; either version 2.1
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# See the files LICENSE.lgpl2 and LICENSE.lgpl3 for full text of the
# GNU Lesser General Public License version 2.1 and 3 respectively.
# This information is also available on the ilastik web site at:
#		   http://ilastik.org/license/
###############################################################################
import numpy
import vigra
from lazyflow.graph import Graph
from lazyflow.operators import OpFilterLabels
from lazyflow.utility.slicingtools import sl, slicing2shape

class TestOpFilterLabels(object):

    def setUp(self):

        inputData = numpy.zeros((1,10,100,100,1), dtype=int)
        inputData = inputData.view(vigra.VigraArray)
        inputData.axistags = vigra.defaultAxistags('txyzc')
        
        inputData[0,0, 50:53, 50:53, 0] = 1 # 9 voxels
        inputData[0,1, 50:52, 50:52, 0] = 2 # 4 voxels
        inputData[0,2, 50:52, 50:53, 0] = 3 # 6 voxels
        
        self.inputData = inputData

    def testBasic(self):
        graph = Graph()
        op = OpFilterLabels(graph=graph)
        op.Input.setValue( self.inputData )
        op.MinLabelSize.setValue(6)

        filtered = op.Output[...].wait()
        assert filtered.shape == self.inputData.shape
        
        expectedData = numpy.array(self.inputData)
        expectedData[0,1, 50:52, 50:52, 0] = 0 # 4 voxels, should be gone
        
        assert (filtered == expectedData).all()
        
        op.MaxLabelSize.setValue(8)
        expectedData[0, 0, 50:53, 50:53, 0] = 0
        filtered2 = op.Output[:].wait()
        assert (filtered2 == expectedData).all()


if __name__ == "__main__":
    import sys
    import nose
    sys.argv.append("--nocapture")    # Don't steal stdout.  Show it on the console as usual.
    sys.argv.append("--nologcapture") # Don't set the logging level to DEBUG.  Leave it alone.
    ret = nose.run(defaultTest=__file__)
    if not ret: sys.exit(1)