DatasetWrapper implements Dataset<C, D> { @Override public void close() throws Exception { delegate.close(); } DatasetWrapper(Dataset<C, D> delegate); @Override R computeWithCtx(IgniteTriFunction<C, D, LearningEnvironment, R> map, IgniteBinaryOperator<R> reduce,
        R identity); @Override R compute(IgniteBiFunction<D, LearningEnvironment, R> map, IgniteBinaryOperator<R> reduce, R identity); @Override void close();  }
@Test public void testClose() throws Exception { wrapper.close(); verify(dataset, times(1)).close(); }