(self, x, axis, expected_avg,
weights, expected_wavg, expected_wsum)
| 365 | [1, 3], [[1.0, 5.0, 9.5]], [[4, 4, 4]])], |
| 366 | ) |
| 367 | def test_basic_keepdims(self, x, axis, expected_avg, |
| 368 | weights, expected_wavg, expected_wsum): |
| 369 | avg = np.average(x, axis=axis, keepdims=True) |
| 370 | assert avg.shape == np.shape(expected_avg) |
| 371 | assert_array_equal(avg, expected_avg) |
| 372 | |
| 373 | wavg = np.average(x, axis=axis, weights=weights, keepdims=True) |
| 374 | assert wavg.shape == np.shape(expected_wavg) |
| 375 | assert_array_equal(wavg, expected_wavg) |
| 376 | |
| 377 | wavg, wsum = np.average(x, axis=axis, weights=weights, returned=True, |
| 378 | keepdims=True) |
| 379 | assert wavg.shape == np.shape(expected_wavg) |
| 380 | assert_array_equal(wavg, expected_wavg) |
| 381 | assert wsum.shape == np.shape(expected_wsum) |
| 382 | assert_array_equal(wsum, expected_wsum) |
| 383 | |
| 384 | def test_weights(self): |
| 385 | y = np.arange(10) |
nothing calls this directly
no test coverage detected