FederatedRepositoryImpl implements FederatedRepository { @Override public FederatedDataSet getCrossValidationData() { return dataSource.getCrossValidationData(); } FederatedRepositoryImpl(FederatedDataSource dataSource, FederatedNetworkDataSource networkDataSource); @Override FederatedDataSet getTrainingData(); @Override FederatedDataSet getTestData(); @Override FederatedDataSet getCrossValidationData(); @Override Observable<Boolean> uploadGradient(byte[] gradient); @Override Observable<byte[]> retrieveGradient(); @Override Observable<Integer> registerModel();  }
@Test public void testGetCrossValidationData() { given(dataSource.getCrossValidationData()).willReturn(dataSet); FederatedDataSet crossValidationData = cut.getCrossValidationData(); assertNotNull(crossValidationData); }