(self)
| 166 | class TestAverage: |
| 167 | # Several tests of average. Why so many ? Good point... |
| 168 | def test_testAverage1(self): |
| 169 | # Test of average. |
| 170 | ott = array([0., 1., 2., 3.], mask=[True, False, False, False]) |
| 171 | assert_equal(2.0, average(ott, axis=0)) |
| 172 | assert_equal(2.0, average(ott, weights=[1., 1., 2., 1.])) |
| 173 | result, wts = average(ott, weights=[1., 1., 2., 1.], returned=True) |
| 174 | assert_equal(2.0, result) |
| 175 | assert_(wts == 4.0) |
| 176 | ott[:] = masked |
| 177 | assert_equal(average(ott, axis=0).mask, [True]) |
| 178 | ott = array([0., 1., 2., 3.], mask=[True, False, False, False]) |
| 179 | ott = ott.reshape(2, 2) |
| 180 | ott[:, 1] = masked |
| 181 | assert_equal(average(ott, axis=0), [2.0, 0.0]) |
| 182 | assert_equal(average(ott, axis=1).mask[0], [True]) |
| 183 | assert_equal([2., 0.], average(ott, axis=0)) |
| 184 | result, wts = average(ott, axis=0, returned=True) |
| 185 | assert_equal(wts, [1., 0.]) |
| 186 | |
| 187 | def test_testAverage2(self): |
| 188 | # More tests of average. |
nothing calls this directly
no test coverage detected