From 9b8762dd88573ca3ec00a6654c76140831e0cde9 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Wed, 22 Aug 2012 15:57:18 +0200
Subject: [PATCH] Fix test import

---
 xbob/db/replay/test.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/xbob/db/replay/test.py b/xbob/db/replay/test.py
index a3aebff..a99664f 100644
--- a/xbob/db/replay/test.py
+++ b/xbob/db/replay/test.py
@@ -22,7 +22,7 @@
 
 import os, sys
 import unittest
-from . import db as replay_db
+from .query import Database
 
 class ReplayDatabaseTest(unittest.TestCase):
   """Performs various tests on the replay attack database."""
@@ -30,7 +30,7 @@ class ReplayDatabaseTest(unittest.TestCase):
   def test01_queryRealAccesses(self):
 
 
-    db = replay_db.Database()
+    db = Database()
     f = db.files(cls='real')
     self.assertEqual(len(set(f.values())), 200) #200 unique auth sessions
     for k,v in f.items():
@@ -53,7 +53,7 @@ class ReplayDatabaseTest(unittest.TestCase):
 
   def queryAttackType(self, protocol, N):
 
-    db = replay_db.Database()
+    db = Database()
     f = db.files(cls='attack', protocol=protocol)
 
     self.assertEqual(len(set(f.values())), N) 
@@ -99,7 +99,7 @@ class ReplayDatabaseTest(unittest.TestCase):
   
   def test08_queryEnrollments(self):
 
-    db = replay_db.Database()
+    db = Database()
     f = db.files(cls='enroll')
     self.assertEqual(len(set(f.values())), 100) #50 clients, 2 conditions
     for k,v in f.items():
-- 
GitLab