(self, arr1d)
| 561 | arr1d.shift(1, fill_value=1) |
| 562 | |
| 563 | def test_median(self, arr1d): |
| 564 | arr = arr1d |
| 565 | if len(arr) % 2 == 0: |
| 566 | # make it easier to define `expected` |
| 567 | arr = arr[:-1] |
| 568 | |
| 569 | expected = arr[len(arr) // 2] |
| 570 | |
| 571 | result = arr.median() |
| 572 | assert type(result) is type(expected) |
| 573 | assert result == expected |
| 574 | |
| 575 | arr[len(arr) // 2] = NaT |
| 576 | if not isinstance(expected, Period): |
| 577 | expected = arr[len(arr) // 2 - 1 : len(arr) // 2 + 2].mean() |
| 578 | |
| 579 | assert arr.median(skipna=False) is NaT |
| 580 | |
| 581 | result = arr.median() |
| 582 | assert type(result) is type(expected) |
| 583 | assert result == expected |
| 584 | |
| 585 | assert arr[:0].median() is NaT |
| 586 | assert arr[:0].median(skipna=False) is NaT |
| 587 | |
| 588 | # 2d Case |
| 589 | arr2 = arr.reshape(-1, 1) |
| 590 | |
| 591 | result = arr2.median(axis=None) |
| 592 | assert type(result) is type(expected) |
| 593 | assert result == expected |
| 594 | |
| 595 | assert arr2.median(axis=None, skipna=False) is NaT |
| 596 | |
| 597 | result = arr2.median(axis=0) |
| 598 | expected2 = type(arr)._from_sequence([expected], dtype=arr.dtype) |
| 599 | tm.assert_equal(result, expected2) |
| 600 | |
| 601 | result = arr2.median(axis=0, skipna=False) |
| 602 | expected2 = type(arr)._from_sequence([NaT], dtype=arr.dtype) |
| 603 | tm.assert_equal(result, expected2) |
| 604 | |
| 605 | result = arr2.median(axis=1) |
| 606 | tm.assert_equal(result, arr) |
| 607 | |
| 608 | result = arr2.median(axis=1, skipna=False) |
| 609 | tm.assert_equal(result, arr) |
| 610 | |
| 611 | def test_from_integer_array(self): |
| 612 | arr = np.array([1, 2, 3], dtype=np.int64) |
nothing calls this directly
no test coverage detected