MCPcopy Create free account
hub / github.com/numpy/numpy / test_edges

Method test_edges

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

Source from the content-addressed store, hash-verified

417 # Tests notmasked_edges and notmasked_contiguous.
418
419 def test_edges(self):
420 # Tests unmasked_edges
421 data = masked_array(np.arange(25).reshape(5, 5),
422 mask=[[0, 0, 1, 0, 0],
423 [0, 0, 0, 1, 1],
424 [1, 1, 0, 0, 0],
425 [0, 0, 0, 0, 0],
426 [1, 1, 1, 0, 0]],)
427 test = notmasked_edges(data, None)
428 assert_equal(test, [0, 24])
429 test = notmasked_edges(data, 0)
430 assert_equal(test[0], [(0, 0, 1, 0, 0), (0, 1, 2, 3, 4)])
431 assert_equal(test[1], [(3, 3, 3, 4, 4), (0, 1, 2, 3, 4)])
432 test = notmasked_edges(data, 1)
433 assert_equal(test[0], [(0, 1, 2, 3, 4), (0, 0, 2, 0, 3)])
434 assert_equal(test[1], [(0, 1, 2, 3, 4), (4, 2, 4, 4, 4)])
435 #
436 test = notmasked_edges(data.data, None)
437 assert_equal(test, [0, 24])
438 test = notmasked_edges(data.data, 0)
439 assert_equal(test[0], [(0, 0, 0, 0, 0), (0, 1, 2, 3, 4)])
440 assert_equal(test[1], [(4, 4, 4, 4, 4), (0, 1, 2, 3, 4)])
441 test = notmasked_edges(data.data, -1)
442 assert_equal(test[0], [(0, 1, 2, 3, 4), (0, 0, 0, 0, 0)])
443 assert_equal(test[1], [(0, 1, 2, 3, 4), (4, 4, 4, 4, 4)])
444 #
445 data[-2] = masked
446 test = notmasked_edges(data, 0)
447 assert_equal(test[0], [(0, 0, 1, 0, 0), (0, 1, 2, 3, 4)])
448 assert_equal(test[1], [(1, 1, 2, 4, 4), (0, 1, 2, 3, 4)])
449 test = notmasked_edges(data, -1)
450 assert_equal(test[0], [(0, 1, 2, 4), (0, 0, 2, 3)])
451 assert_equal(test[1], [(0, 1, 2, 4), (4, 2, 4, 4)])
452
453 def test_contiguous(self):
454 # 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