@DataProvider(name = "subSeries") public static Object[][] data_subSeries() { return new Object[][] { // start = end -> empty {DATE_2011_01_01, DATE_2011_01_01, new int[] {}}, // no overlap {date(2006, 1, 1), date(2009, 1, 1), new int[] {}}, // single point {DATE_2015_01_06, DATE_2015_01_07, new int[] {1}}, // include when start matches base, exclude when end matches base {DATE_2015_01_06, DATE_2015_01_08, new int[] {1, 2}}, // include when start matches base {DATE_2015_01_05, DATE_2015_01_09, new int[] {0, 1, 2, 3}}, // neither start nor end match {date(2014, 12, 31), date(2015, 2, 1), new int[] {0, 1, 2, 3, 4}}, }; }
public void test_mapDates() { List<Double> values = values(1, 2, 4, 5, 8); LocalDateDoubleTimeSeries base = LocalDateDoubleTimeSeries.builder().putAll(DATES_2015_1_WEEK, values).build(); LocalDateDoubleTimeSeries test = base.mapDates(date -> date.plusYears(1)); ImmutableList<LocalDate> expectedDates = ImmutableList.of(date(2016, 1, 5), date(2016, 1, 6), date(2016, 1, 7), date(2016, 1, 8), date(2016, 1, 9)); LocalDateDoubleTimeSeries expected = LocalDateDoubleTimeSeries.builder().putAll(expectedDates, values).build(); assertEquals(test, expected); }
@Test(expectedExceptions = IllegalArgumentException.class) public void test_subSeries_startAfterEnd() { LocalDateDoubleTimeSeries base = LocalDateDoubleTimeSeries.builder().putAll(DATES_2015_1_WEEK, VALUES_1_WEEK).build(); base.subSeries(date(2011, 1, 2), DATE_2011_01_01); }
public void test_mapDates_notAscending() { List<Double> values = values(1, 2, 4, 5, 8); LocalDateDoubleTimeSeries base = LocalDateDoubleTimeSeries.builder().putAll(DATES_2015_1_WEEK, values).build(); assertThrowsIllegalArg(() -> base.mapDates(date -> date(2016, 1, 6))); }
public void test_subSeries_picks_valid_dates() { LocalDateDoubleTimeSeries series = LocalDateDoubleTimeSeries.builder() .put(DATE_2015_01_02, 10) // Friday .put(DATE_2015_01_05, 11) // Mon .put(DATE_2015_01_06, 12) .put(DATE_2015_01_07, 13) .put(DATE_2015_01_08, 14) .put(DATE_2015_01_09, 15) // Fri .put(DATE_2015_01_12, 16) // Mon .build(); // Pick using weekend dates LocalDateDoubleTimeSeries subSeries = series.subSeries(DATE_2015_01_04, date(2015, 1, 10)); assertEquals(subSeries.size(), 5); assertEquals(subSeries.get(DATE_2015_01_02), OptionalDouble.empty()); assertEquals(subSeries.get(DATE_2015_01_04), OptionalDouble.empty()); assertEquals(subSeries.get(DATE_2015_01_05), OptionalDouble.of(11)); assertEquals(subSeries.get(DATE_2015_01_06), OptionalDouble.of(12)); assertEquals(subSeries.get(DATE_2015_01_07), OptionalDouble.of(13)); assertEquals(subSeries.get(DATE_2015_01_08), OptionalDouble.of(14)); assertEquals(subSeries.get(DATE_2015_01_09), OptionalDouble.of(15)); assertEquals(subSeries.get(DATE_2015_01_12), OptionalDouble.empty()); }