def test_duplicate_name(self, load, is_valid_cube):
cube1 = Mock(**{'name.return_value': 'foo'})
cube2 = Mock(**{'name.return_value': 'foo'})
load.return_value = [cube1, cube2]
is_valid_cube.return_value = True
result = gridded_forecast._load(sentinel.pattern)
load.assert_called_once_with(sentinel.pattern)
self.assertEqual(is_valid_cube.mock_calls, [call(cube1), call(cube2)])
self.assertEqual(result, {'foo (1)': cube1, 'foo (2)': cube2})
def test_none_valid(self, load, is_valid_cube):
load.return_value = ['foo', 'bar']
is_valid_cube.return_value = False
with self.assertRaises(AssertionError):
gridded_forecast._load(None)