Fixing tests

runtests.sh

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

tests/test_Handshake.py

@@ -4,7 +4,7 @@
 import sys
 sys.path.insert(0, '..')
 
-from py.Handshake import Handshake
+from wifite.model.handshake import Handshake
 
 import unittest
 

tests/test_Target.py

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