def test_edges(self):
     # Tests unmasked_edges
     data = masked_array(
         np.arange(25).reshape(5, 5),
         mask=[[0, 0, 1, 0, 0], [0, 0, 0, 1, 1], [1, 1, 0, 0, 0],
               [0, 0, 0, 0, 0], [1, 1, 1, 0, 0]],
     )
     test = notmasked_edges(data, None)
     assert_equal(test, [0, 24])
     test = notmasked_edges(data, 0)
     assert_equal(test[0], [(0, 0, 1, 0, 0), (0, 1, 2, 3, 4)])
     assert_equal(test[1], [(3, 3, 3, 4, 4), (0, 1, 2, 3, 4)])
     test = notmasked_edges(data, 1)
     assert_equal(test[0], [(0, 1, 2, 3, 4), (0, 0, 2, 0, 3)])
     assert_equal(test[1], [(0, 1, 2, 3, 4), (4, 2, 4, 4, 4)])
     #
     test = notmasked_edges(data.data, None)
     assert_equal(test, [0, 24])
     test = notmasked_edges(data.data, 0)
     assert_equal(test[0], [(0, 0, 0, 0, 0), (0, 1, 2, 3, 4)])
     assert_equal(test[1], [(4, 4, 4, 4, 4), (0, 1, 2, 3, 4)])
     test = notmasked_edges(data.data, -1)
     assert_equal(test[0], [(0, 1, 2, 3, 4), (0, 0, 0, 0, 0)])
     assert_equal(test[1], [(0, 1, 2, 3, 4), (4, 4, 4, 4, 4)])
     #
     data[-2] = masked
     test = notmasked_edges(data, 0)
     assert_equal(test[0], [(0, 0, 1, 0, 0), (0, 1, 2, 3, 4)])
     assert_equal(test[1], [(1, 1, 2, 4, 4), (0, 1, 2, 3, 4)])
     test = notmasked_edges(data, -1)
     assert_equal(test[0], [(0, 1, 2, 4), (0, 0, 2, 3)])
     assert_equal(test[1], [(0, 1, 2, 4), (4, 2, 4, 4)])
Example #2
0
 def test_edges(self):
     # Tests unmasked_edges
     data = masked_array(np.arange(25).reshape(5, 5),
                         mask=[[0, 0, 1, 0, 0],
                               [0, 0, 0, 1, 1],
                               [1, 1, 0, 0, 0],
                               [0, 0, 0, 0, 0],
                               [1, 1, 1, 0, 0]],)
     test = notmasked_edges(data, None)
     assert_equal(test, [0, 24])
     test = notmasked_edges(data, 0)
     assert_equal(test[0], [(0, 0, 1, 0, 0), (0, 1, 2, 3, 4)])
     assert_equal(test[1], [(3, 3, 3, 4, 4), (0, 1, 2, 3, 4)])
     test = notmasked_edges(data, 1)
     assert_equal(test[0], [(0, 1, 2, 3, 4), (0, 0, 2, 0, 3)])
     assert_equal(test[1], [(0, 1, 2, 3, 4), (4, 2, 4, 4, 4)])
     #
     test = notmasked_edges(data.data, None)
     assert_equal(test, [0, 24])
     test = notmasked_edges(data.data, 0)
     assert_equal(test[0], [(0, 0, 0, 0, 0), (0, 1, 2, 3, 4)])
     assert_equal(test[1], [(4, 4, 4, 4, 4), (0, 1, 2, 3, 4)])
     test = notmasked_edges(data.data, -1)
     assert_equal(test[0], [(0, 1, 2, 3, 4), (0, 0, 0, 0, 0)])
     assert_equal(test[1], [(0, 1, 2, 3, 4), (4, 4, 4, 4, 4)])
     #
     data[-2] = masked
     test = notmasked_edges(data, 0)
     assert_equal(test[0], [(0, 0, 1, 0, 0), (0, 1, 2, 3, 4)])
     assert_equal(test[1], [(1, 1, 2, 4, 4), (0, 1, 2, 3, 4)])
     test = notmasked_edges(data, -1)
     assert_equal(test[0], [(0, 1, 2, 4), (0, 0, 2, 3)])
     assert_equal(test[1], [(0, 1, 2, 4), (4, 2, 4, 4)])