Add simple unit test

setup.py

@@ -45,6 +45,18 @@ setup(
       'xbob.db',
     ],
 
+    entry_points = {
+      # declare database to bob
+      #'bob.db': [
+      #  'subvoxforge = xbob.db.subvoxforge.driver:Interface',
+      #  ],
+
+      # declare tests to bob
+      'bob.test': [
+        'subvoxforge = xbob.db.subvoxforge.test:SubvoxforgeDatabaseTest',
+        ],
+      },
+
     classifiers = [
       'Development Status :: 4 - Beta',
       'Intended Audience :: Education',

xbob/db/subvoxforge/test.py

+#!/usr/bin/env python
+# vim: set fileencoding=utf-8 :
+# Laurent El Shafey <laurent.el-shafey@idiap.ch>
+# Fri Aug 23 16:27:27 CEST 2013
+#
+# Copyright (C) 2011-2012 Idiap Research Institute, Martigny, Switzerland
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, version 3 of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+"""A few checks on the protocols of a subset of the VoxForge database
+"""
+
+import os, sys
+import unittest
+from .query import Database
+
+class SubvoxforgeDatabaseTest(unittest.TestCase):
+  """Performs various tests on the protocols of a subset of the VoxForge database."""
+
+  def test01_query(self):
+    from pkg_resources import resource_filename
+    #db = Database(example_data, use_dense_probe_file_list = False)
+    db = Database()
+
+    self.assertEqual(len(db.client_ids()), 30) # 10 client ids for world, dev and eval
+    self.assertEqual(len(db.client_ids(groups='world')), 10) # 10 client ids for world
+    self.assertEqual(len(db.client_ids(groups='optional_world_1')), 10) # 10 client ids for optional world 1
+    self.assertEqual(len(db.client_ids(groups='optional_world_2')), 10) # 10 client ids for optional world 2
+    self.assertEqual(len(db.client_ids(groups='dev')), 10) # 10 client ids for dev
+    self.assertEqual(len(db.client_ids(groups='eval')), 10) # 10 client ids for eval
+
+    self.assertEqual(len(db.model_ids()), 30) # 30 model ids for world, dev and eval
+    self.assertEqual(len(db.model_ids(groups='world')), 10) # 10 model ids for world
+    self.assertEqual(len(db.model_ids(groups='optional_world_1')), 10) # 10 model ids for optional world 1
+    self.assertEqual(len(db.model_ids(groups='optional_world_2')), 10) # 10 model ids for optional world 2
+    self.assertEqual(len(db.model_ids(groups='dev')), 10) # 10 model ids for dev
+    self.assertEqual(len(db.model_ids(groups='eval')), 10) # 10 model ids for eval
+
+    self.assertEqual(len(db.objects(groups='world')), 3148) # 3148 samples in the world set
+
+    self.assertEqual(len(db.objects(groups='dev', purposes='enrol')), 1304) # 1304 samples for enrollment in the dev set
+    self.assertEqual(len(db.objects(groups='dev', purposes='enrol', model_ids='Dcoetzee')), 240) # 240 samples to enroll model 'Dcoetzee' in the dev set
+    self.assertEqual(len(db.objects(groups='dev', purposes='enrol', model_ids='rortiz')), 0) # 0 samples to enroll model 'rortiz' (it is an eval model)
+    self.assertEqual(len(db.objects(groups='dev', purposes='probe')), 300) # 300 samples as probes in the dev set
+
+    self.assertEqual(len(db.objects(groups='eval', purposes='enrol')), 1509) # 1509 samples for enrollment in the eval set
+    self.assertEqual(len(db.objects(groups='eval', purposes='enrol', model_ids='rortiz')), 120) # 120 samples to enroll model 'rortiz' in the eval set
+    self.assertEqual(len(db.objects(groups='eval', purposes='enrol', model_ids='Dcoetzee')), 0) # 0 samples to enroll model 'Dcoetzee' (it is a dev model)
+    self.assertEqual(len(db.objects(groups='eval', purposes='probe')), 300) # 300 samples as probes in the eval set
+