Skip to content
Snippets Groups Projects
Commit 521a44a7 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Fix the dummy database

parent a84d12d9
No related branches found
No related tags found
No related merge requests found
Pipeline #
from bob.bio.db import ZTBioDatabase, AtntBioDatabase from bob.bio.db import ZTBioDatabase
from bob.bio.base.test.utils import atnt_database_directory from bob.bio.base.test.utils import atnt_database_directory
...@@ -14,13 +14,14 @@ class DummyDatabase(ZTBioDatabase): ...@@ -14,13 +14,14 @@ class DummyDatabase(ZTBioDatabase):
training_depends_on_protocol=False, training_depends_on_protocol=False,
models_depend_on_protocol=False models_depend_on_protocol=False
) )
self.__db = AtntBioDatabase() import bob.db.atnt
self.__db = bob.db.atnt.Database()
def model_ids(self, groups=None, protocol=None, gender=None): def model_ids_with_protocol(self, groups=None, protocol=None, **kwargs):
return self.__db.model_ids(groups, protocol, gender) return self.__db.model_ids(groups, protocol)
def objects(self, groups=None, protocol=None, purposes=None, model_ids=None, **kwargs): def objects(self, groups=None, protocol=None, purposes=None, model_ids=None, **kwargs):
return self.__db.objects(groups, protocol, purposes, model_ids, **kwargs) return self.__db.objects(model_ids, groups, purposes, protocol, **kwargs)
def tobjects(self, groups=None, protocol=None, model_ids=None, **kwargs): def tobjects(self, groups=None, protocol=None, model_ids=None, **kwargs):
return [] return []
...@@ -28,13 +29,13 @@ class DummyDatabase(ZTBioDatabase): ...@@ -28,13 +29,13 @@ class DummyDatabase(ZTBioDatabase):
def zobjects(self, groups=None, protocol=None, **kwargs): def zobjects(self, groups=None, protocol=None, **kwargs):
return [] return []
def tmodel_ids(self, protocol=None, groups=None, **kwargs): def tmodel_ids_with_protocol(self, protocol=None, groups=None, **kwargs):
return self.__db.model_ids(groups) return self.__db.model_ids(groups)
def t_enroll_files(self, t_model_id, group='dev'): def t_enroll_files(self, t_model_id, group='dev'):
return self.__db.enroll_files(t_model_id, group) return self.enroll_files(t_model_id, group)
def z_probe_files(self, group='dev'): def z_probe_files(self, group='dev'):
return self.__db.probe_files(None, group) return self.probe_files(None, group)
database = DummyDatabase() database = DummyDatabase()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment