Fix tests for py3.

This commit is contained in:
Steve Pulec 2017-02-23 22:28:09 -05:00
commit 3c0c4c2996
10 changed files with 16 additions and 20 deletions

View file

@ -62,10 +62,10 @@ BACKENDS = {
}
def get_model(name, region):
def get_model(name, region_name):
for backends in BACKENDS.values():
for region, backend in backends.items():
if region == region:
if region == region_name:
models = getattr(backend.__class__, '__models__', {})
if name in models:
return list(getattr(backend, models[name])())