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)