From 742401f61b3782b287386208fdd1f1d5d14f88df Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Thu, 30 Nov 2017 11:49:11 +0100
Subject: [PATCH] Add tests for flat=True
---
bob/pad/base/test/test_databases.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/bob/pad/base/test/test_databases.py b/bob/pad/base/test/test_databases.py
index 3653eb5..703c987 100644
--- a/bob/pad/base/test/test_databases.py
+++ b/bob/pad/base/test/test_databases.py
@@ -43,6 +43,10 @@ class DummyDatabaseSqlTest(unittest.TestCase):
check_file(db.training_files(), 2)
check_file(db.files([1]))
check_file(db.reverse(["test/path"]))
+ # check if flat returns flat files
+ assert len(db.all_files(flat=True)) == 2, db.all_files(flat=True)
+ check_file(db.all_files(flat=True)[0:1])
+ check_file(db.all_files(flat=True)[1:2])
file = db.objects()[0]
assert db.original_file_name(file) == "original/directory/test/path.orig"
--
GitLab