@Test
@SuppressWarnings("unchecked")
public void getPaths() {

    // Mock options
    final DataSetOptions options = new DataSetOptions();
    options.setFormat("mock");
    options.setOption(KyloCatalogConstants.PATH_OPTION, "/mock/path/file1.txt");
    options.setPaths(Collections.singletonList("/mock/path/file2.txt"));
    // Test getting path for non-file-format data sources
    final SparkDataSetContext<DataFrame> context = new SparkDataSetContext<>(options, Mockito.mock(KyloCatalogClient.class), Mockito.mock(SparkDataSetDelegate.class));
    Assert.assertEquals("/mock/path/file1.txt", context.getOption("path").get());
    Assert.assertEquals(Collections.singletonList("/mock/path/file2.txt"), context.getPaths());
}