(self, f)
| 108 | assert_equal(test, control) |
| 109 | |
| 110 | def check_clump(self, f): |
| 111 | for i in range(1, 7): |
| 112 | for j in range(2**i): |
| 113 | k = np.arange(i, dtype=int) |
| 114 | ja = np.full(i, j, dtype=int) |
| 115 | a = masked_array(2**k) |
| 116 | a.mask = (ja & (2**k)) != 0 |
| 117 | s = 0 |
| 118 | for sl in f(a): |
| 119 | s += a.data[sl].sum() |
| 120 | if f == clump_unmasked: |
| 121 | assert_equal(a.compressed().sum(), s) |
| 122 | else: |
| 123 | a.mask = ~a.mask |
| 124 | assert_equal(a.compressed().sum(), s) |
| 125 | |
| 126 | def test_clump_masked(self): |
| 127 | # Test clump_masked |
no test coverage detected