def test_poly2mask(): poly1 = [[0, 0], [0, 1], [1, 1], [1, 0]] poly2 = [[0, 1], [0, 2], [2, 2], [2, 1]] masks = ROI.poly2mask([poly1, poly2], (3, 3)) assert_equal( masks[0].todense(), np.matrix([[True, False, False], [True, True, False], [False, False, False]], dtype=bool))
def test_poly2mask(): poly1 = [[0, 0], [0, 1], [1, 1], [1, 0]] poly2 = [[0, 1], [0, 2], [2, 2], [2, 1]] masks = ROI.poly2mask([poly1, poly2], (3, 3)) assert_equal( masks[0].todense(), np.matrix( [[True, False, False], [True, True, False], [False, False, False]], dtype=bool))