MCPcopy Index your code
hub / github.com/numpy/numpy / test_edges

Method test_edges

numpy/ma/tests/test_extras.py:507–539  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

505 # Tests notmasked_edges and notmasked_contiguous.
506
507 def test_edges(self):
508 # Tests unmasked_edges
509 data = masked_array(np.arange(25).reshape(5, 5),
510 mask=[[0, 0, 1, 0, 0],
511 [0, 0, 0, 1, 1],
512 [1, 1, 0, 0, 0],
513 [0, 0, 0, 0, 0],
514 [1, 1, 1, 0, 0]],)
515 test = notmasked_edges(data, None)
516 assert_equal(test, [0, 24])
517 test = notmasked_edges(data, 0)
518 assert_equal(test[0], [(0, 0, 1, 0, 0), (0, 1, 2, 3, 4)])
519 assert_equal(test[1], [(3, 3, 3, 4, 4), (0, 1, 2, 3, 4)])
520 test = notmasked_edges(data, 1)
521 assert_equal(test[0], [(0, 1, 2, 3, 4), (0, 0, 2, 0, 3)])
522 assert_equal(test[1], [(0, 1, 2, 3, 4), (4, 2, 4, 4, 4)])
523 #
524 test = notmasked_edges(data.data, None)
525 assert_equal(test, [0, 24])
526 test = notmasked_edges(data.data, 0)
527 assert_equal(test[0], [(0, 0, 0, 0, 0), (0, 1, 2, 3, 4)])
528 assert_equal(test[1], [(4, 4, 4, 4, 4), (0, 1, 2, 3, 4)])
529 test = notmasked_edges(data.data, -1)
530 assert_equal(test[0], [(0, 1, 2, 3, 4), (0, 0, 0, 0, 0)])
531 assert_equal(test[1], [(0, 1, 2, 3, 4), (4, 4, 4, 4, 4)])
532 #
533 data[-2] = masked
534 test = notmasked_edges(data, 0)
535 assert_equal(test[0], [(0, 0, 1, 0, 0), (0, 1, 2, 3, 4)])
536 assert_equal(test[1], [(1, 1, 2, 4, 4), (0, 1, 2, 3, 4)])
537 test = notmasked_edges(data, -1)
538 assert_equal(test[0], [(0, 1, 2, 4), (0, 0, 2, 3)])
539 assert_equal(test[1], [(0, 1, 2, 4), (4, 2, 4, 4)])
540
541 def test_contiguous(self):
542 # Tests notmasked_contiguous

Callers

nothing calls this directly

Calls 3

notmasked_edgesFunction · 0.90
assert_equalFunction · 0.90
reshapeMethod · 0.80

Tested by

no test coverage detected