Fixing unit tests

py/Configuration.py

@@ -7,6 +7,7 @@ import os
 
 class Configuration(object):
     ''' Stores configuration variables and functions for Wifite. '''
+    verbose = 0
 
     initialized = False # Flag indicating config has been initialized
     temp_dir = None     # Temporary directory

py/tests/test_Handshake.py

@@ -4,7 +4,7 @@
 import sys
 sys.path.insert(0, '..')
 
-from Handshake import Handshake
+from py.Handshake import Handshake
 
 import unittest
 
@@ -20,7 +20,6 @@ class TestHandshake(unittest.TestCase):
 
     def testAnalyze(self):
         hs_file = self.getFile('handshake_exists.cap')
-        print hs_file
         hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
         try:
             hs.analyze()

py/tests/test_Target.py

@@ -1,7 +1,7 @@
 #!/usr/bin/python2.7
 # -*- coding: utf-8 -*-
 
-from Airodump import Airodump
+from py.Airodump import Airodump
 
 import unittest
 

runtests.sh

@@ -1,2 +1,2 @@
 #!/bin/sh
-python2.7 -m unittest discover py
+python2.7 -m unittest discover py/tests -v