@Test
public void testGetTrainingData() {

    // Given
    given(dataSource.getTrainingData()).willReturn(dataSet);
    // When
    FederatedDataSet trainingData = cut.getTrainingData();
    // Then
    assertNotNull(trainingData);
}