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)