def file_loader(file_type, pattern, label=None, locator=None):
file_type = file_type.lower().replace("_", "")
if file_type == 'rdt':
return rdt.Loader(pattern)
elif file_type == 'gpm':
return data.GPM(pattern)
elif file_type == 'earthnetworks':
return earth_networks.Loader.pattern(pattern)
elif file_type == 'eida50':
return satellite.EIDA50(pattern)
elif file_type == 'griddedforecast':
return gridded_forecast.ImageLoader(label, pattern)
elif file_type == 'unifiedmodel':
return data.DBLoader(label, pattern, locator)
else:
raise Exception("unrecognised file_type: {}".format(file_type))
def test_init(self, load):
load.return_value = sentinel.cubes
result = gridded_forecast.ImageLoader(sentinel.label, sentinel.pattern)
load.assert_called_once_with(sentinel.pattern)
self.assertEqual(result._label, sentinel.label)
self.assertEqual(result._cubes, sentinel.cubes)