diff --git a/bob/pad/base/test/dummy/database.py b/bob/pad/base/test/dummy/database.py index a48c301c525bcccc2491aee7776395476742aec1..adb5e884f62f32e0e96770cdd0da91b17b66b573 100644 --- a/bob/pad/base/test/dummy/database.py +++ b/bob/pad/base/test/dummy/database.py @@ -29,11 +29,13 @@ dummy_data = {'train_real': 1.0, 'train_attack': 2.0, class TestFile(PadFile): + def __init__(self, path, id): attack_type = None if "attack" in path: attack_type = "attack" - PadFile.__init__(self, client_id=1, path=path, file_id=id, attack_type=attack_type) + PadFile.__init__(self, client_id=1, path=path, + file_id=id, attack_type=attack_type) def load(self, directory=None, extension='.hdf5'): """Loads the data at the specified location and using the given extension. @@ -57,6 +59,7 @@ class TestFile(PadFile): path = self.make_path(directory or '', extension or '') return dummy_data[os.path.basename(path)] + def dumplist(args): """Dumps lists of files based on your criteria""" @@ -76,6 +79,7 @@ def dumplist(args): class Interface(BaseInterface): + def name(self): return dummy_name @@ -96,11 +100,11 @@ class Interface(BaseInterface): dumpparser = subparsers.add_parser('dumplist', help="") dumpparser.add_argument('-d', '--directory', dest="directory", default='', - help="if given, this path will be prepended to every entry returned (defaults to '%(default)s')") + help="if given, this path will be prepended to every entry returned (defaults to '%(default)s')") dumpparser.add_argument('-e', '--extension', dest="extension", default='', - help="if given, this extension will be appended to every entry returned (defaults to '%(default)s')") + help="if given, this extension will be appended to every entry returned (defaults to '%(default)s')") dumpparser.add_argument('--self-test', dest="selftest", default=False, - action='store_true', help=SUPPRESS) + action='store_true', help=SUPPRESS) dumpparser.set_defaults(func=dumplist) # action @@ -110,9 +114,9 @@ class TestDatabase(PadDatabase): def __init__(self, protocol='Default', original_directory=data_dir, original_extension='', **kwargs): # call base class constructors to open a session to the database - PadDatabase.__init__(self, name='testspoof', protocol=protocol, - original_directory=original_directory, - original_extension=original_extension, **kwargs) + super(TestDatabase, self).__init__(name='testspoof', protocol=protocol, + original_directory=original_directory, + original_extension=original_extension, **kwargs) ################################################ # Low level support methods for the database # @@ -168,6 +172,7 @@ class TestDatabase(PadDatabase): # does not implement the given access protocol return False + def get_all_data(self): return self.all_files() diff --git a/bob/pad/base/test/dummy/database_sql.py b/bob/pad/base/test/dummy/database_sql.py index 0d9e10105cc34f8375f460fadfd64818a84ae032..9c9d0f8fdab94f3df4279800029fa1d784f206a1 100644 --- a/bob/pad/base/test/dummy/database_sql.py +++ b/bob/pad/base/test/dummy/database_sql.py @@ -45,10 +45,10 @@ def create_database(): del engine -class TestDatabaseSql (bob.pad.base.database.PadDatabase, bob.db.base.SQLiteDatabase): +class TestDatabaseSql (bob.pad.base.database.PadDatabase, bob.db.base.SQLiteBaseDatabase): def __init__(self): bob.pad.base.database.PadDatabase.__init__(self, 'pad_test', original_directory="original/directory", original_extension=".orig") - bob.db.base.SQLiteDatabase.__init__(self, dbfile, TestFileSql) + bob.db.base.SQLiteBaseDatabase.__init__(self, dbfile, TestFileSql) def groups(self, protocol=None): return ['group'] @@ -59,4 +59,4 @@ class TestDatabaseSql (bob.pad.base.database.PadDatabase, bob.db.base.SQLiteData def annotations(self, file): return None -database = TestDatabaseSql() \ No newline at end of file +database = TestDatabaseSql()