DateTimeColumnTest.java

/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package tech.tablesaw.api;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertTrue;

import java.time.LocalDateTime;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;
import java.util.List;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import tech.tablesaw.columns.datetimes.DateTimeParser;
import tech.tablesaw.columns.strings.StringColumnType;

public class DateTimeColumnTest {

  private DateTimeColumn column1;

  @BeforeEach
  public void setUp() {
    Table table = Table.create("Test");
    column1 = DateTimeColumn.create("Game date");
    table.addColumns(column1);
  }

  @Test
  public void testAppendCell() {
    column1.appendCell("1923-10-20T10:15:30");
    column1.appendCell("1924-12-10T10:15:30");
    column1.appendCell("2015-12-05T10:15:30");
    column1.appendCell("2015-12-20T10:15:30");
    assertEquals(4, column1.size());
    LocalDateTime date = LocalDateTime.now();
    column1.append(date);
    assertEquals(5, column1.size());
  }

  @Test
  public void testAppendCell2() {
    column1.appendCell("10/12/2016 12:18:03 AM");
    column1.appendCell("10/2/2016 8:18:03 AM");
    column1.appendCell("10/12/2016 12:18:03 AM");
    assertEquals(3, column1.size());
  }

  @Test
  public void testCustomParser() {
    // Just do enough to ensure the parser is wired up correctly
    DateTimeParser customParser = new DateTimeParser(ColumnType.LOCAL_DATE_TIME);
    customParser.setMissingValueStrings(List.of("not here"));
    column1.setParser(customParser);

    column1.appendCell("not here");
    assertTrue(column1.isMissing(column1.size() - 1));
    column1.appendCell("1923-10-20T10:15:30");
    assertFalse(column1.isMissing(column1.size() - 1));
  }

  @Test
  public void testConvertMillisSinceEpoch() {
    long millis = 1503952123189L;
    LongColumn dc = LongColumn.create("test");
    dc.append(millis);
    DateTimeColumn column2 = dc.asDateTimes(ZoneOffset.UTC);

    assertEquals(1, column2.size());
    assertEquals(2017, column2.get(0).getYear());
    assertEquals(8, column2.get(0).getMonthValue());
    assertEquals(28, column2.get(0).getDayOfMonth());
    assertEquals(20, column2.get(0).getHour());

    long[] millisArr = column2.asEpochMillisArray();
    assertEquals(1, millisArr.length);
    assertEquals(millis, millisArr[0]);
  }

  @Test
  public void testAfter() {
    Table t = Table.create("test");
    t.addColumns(column1);
    column1.appendCell("2015-12-03T10:15:30");
    column1.appendCell("2015-01-03T10:15:30");
    Table result =
        t.where(t.dateTimeColumn("Game date").isAfter(LocalDateTime.of(2015, 2, 2, 0, 0)));
    assertEquals(1, result.rowCount());
  }

  @Test
  public void testNull() {
    DateTimeColumn col = DateTimeColumn.create("Game date");
    col.appendCell(null);
    assertNull(col.get(0));
  }

  @Test
  public void testCountUnique() {
    column1.append(LocalDateTime.of(2000, 1, 1, 0, 0));
    column1.append(LocalDateTime.of(2000, 1, 1, 0, 0));
    column1.append(LocalDateTime.of(2000, 2, 1, 0, 0));
    column1.appendMissing();

    assertEquals(3, column1.countUnique());
  }

  @Test
  public void testFormatter() {
    column1.setPrintFormatter(DateTimeFormatter.ISO_LOCAL_DATE_TIME, "NaT");
    column1.append(LocalDateTime.of(2000, 1, 1, 0, 0));
    column1.appendMissing();
    assertEquals("2000-01-01T00:00:00", column1.getString(0));
    assertEquals("NaT", column1.getString(1));
  }

  @Test
  public void testAsStringColumn() {
    column1.appendCell("1923-10-20T10:15:30");
    column1.appendMissing();
    StringColumn sc = column1.asStringColumn();
    assertEquals("Game date strings", sc.name());
    assertEquals(2, sc.size());
    assertEquals("1923-10-20T10:15:30.000", sc.get(0));
    assertEquals(StringColumnType.missingValueIndicator(), sc.get(1));
  }
}