def METHOD_NAME(self):
    """ Testing the method load train partition """
    # Test with split
    self.torch_data_manager.split(0.5)
    loader = self.torch_data_manager.load_all_samples()
    self.assertEqual(len(loader.dataset), len(self.dataset), 'Did not properly get loader for all samples')