@Test
public void testFitBatchTable() {

    Long id = MLEnvironmentFactory.getNewMLEnvironmentId();
    MLEnvironment env = MLEnvironmentFactory.get(id);
    DataSet<Integer> input= env.getExecutionEnvironment().fromElements(1, 2, 3);
    Table table = env.getBatchTableEnvironment().fromDataSet(input);
    FakeEstimator estimator = new FakeEstimator();
    estimator.setMLEnvironmentId(id);
    estimator.fit(env.getBatchTableEnvironment(), table);
    Assert.assertTrue(estimator.batchFitted);
    Assert.assertFalse(estimator.streamFitted);
}