|
15 | 15 | from pandas import (
|
16 | 16 | Index,
|
17 | 17 | Period,
|
| 18 | + PeriodIndex, |
18 | 19 | Series,
|
19 | 20 | Timestamp,
|
| 21 | + arrays, |
20 | 22 | date_range,
|
21 | 23 | )
|
22 | 24 | import pandas._testing as tm
|
@@ -375,3 +377,32 @@ def get_view_interval(self):
|
375 | 377 | tdc = converter.TimeSeries_TimedeltaFormatter()
|
376 | 378 | monkeypatch.setattr(tdc, "axis", mock_axis())
|
377 | 379 | tdc(0.0, 0)
|
| 380 | + |
| 381 | + |
| 382 | +@pytest.mark.parametrize("year_span", [11.25, 30, 80, 150, 400, 800, 1500, 2500, 3500]) |
| 383 | +# The range is limited to 11.25 at the bottom by if statements in |
| 384 | +# the _quarterly_finder() function |
| 385 | +def test_quarterly_finder(year_span): |
| 386 | + vmin = -1000 |
| 387 | + vmax = vmin + year_span * 4 |
| 388 | + span = vmax - vmin + 1 |
| 389 | + if span < 45: # the quarterly finder is only invoked if the span is >= 45 |
| 390 | + return |
| 391 | + nyears = span / 4 |
| 392 | + (min_anndef, maj_anndef) = converter._get_default_annual_spacing(nyears) |
| 393 | + result = converter._quarterly_finder(vmin, vmax, "Q") |
| 394 | + quarters = PeriodIndex( |
| 395 | + arrays.PeriodArray(np.array([x[0] for x in result]), freq="Q") |
| 396 | + ) |
| 397 | + majors = np.array([x[1] for x in result]) |
| 398 | + minors = np.array([x[2] for x in result]) |
| 399 | + major_quarters = quarters[majors] |
| 400 | + minor_quarters = quarters[minors] |
| 401 | + check_major_years = major_quarters.year % maj_anndef == 0 |
| 402 | + check_minor_years = minor_quarters.year % min_anndef == 0 |
| 403 | + check_major_quarters = major_quarters.quarter == 1 |
| 404 | + check_minor_quarters = minor_quarters.quarter == 1 |
| 405 | + assert np.all(check_major_years) |
| 406 | + assert np.all(check_minor_years) |
| 407 | + assert np.all(check_major_quarters) |
| 408 | + assert np.all(check_minor_quarters) |
0 commit comments