commit e8ed2b3eb44139da4adaae2fc46c45d8627f7a30
author Ang Li <angli@google.com> Fri Dec 11 12:30:20 2015 -0800
committer Ang Li <angli@google.com> Fri Dec 11 13:20:43 2015 -0800
tree fe5cfc1dfd3667de613c592c59f506e0a86289dc
parent ccf2a90fa8e72aa3c3630277560a5b6f4ce7e73a

Fix import erros in Wifi scripts.

Change-Id: I58f373fc9f0f151193d973980d5849a7675a92de

acts/tests/google/wifi/WifiEnterpriseTest.py

diff --git a/acts/tests/google/wifi/WifiEnterpriseTest.py b/acts/tests/google/wifi/WifiEnterpriseTest.py
index 625926c..8fef2db 100755
--- a/acts/tests/google/wifi/WifiEnterpriseTest.py
+++ b/acts/tests/google/wifi/WifiEnterpriseTest.py
@@ -19,9 +19,10 @@
 import time
 
 import acts.base_test
-import acts.signals.generated_test as generated_test
+import acts.signals
 import acts.test_utils.wifi_test_utils as wutils
-import acts.test_utils.wifi_test_utils.WifiEnums as WifiEnums
+
+WifiEnums = wutils.WifiEnums
 
 # EAP Macros
 EAP = WifiEnums.Eap
@@ -33,7 +34,7 @@
 class WifiEnterpriseTest(acts.base_test.BaseTestClass):
 
     def __init__(self, controllers):
-        BaseTestClass.__init__(self, controllers)
+        acts.base_test.BaseTestClass.__init__(self, controllers)
         self.tests = (
             "test_eap_connect",
             "test_eap_connect_negative",
@@ -315,7 +316,7 @@
         return name
 
     """Tests"""
-    @generated_test
+    @acts.signals.generated_test
     def test_eap_connect(self):
         """Test connecting to enterprise networks of different authentication
         types.
@@ -346,7 +347,7 @@
                pprint.pformat(failed))
         self.assert_true(len(failed) == 0, msg)
 
-    @generated_test
+    @acts.signals.generated_test
     def test_eap_connect_negative(self):
         """Test connecting to enterprise networks.
 
@@ -373,7 +374,7 @@
                pprint.pformat(failed))
         self.assert_true(len(failed) == 0, msg)
 
-    @generated_test
+    @acts.signals.generated_test
     def test_passpoint_connect(self):
         """Test connecting to enterprise networks of different authentication
         types with passpoint support.
@@ -405,7 +406,7 @@
                pprint.pformat(failed))
         self.assert_true(len(failed) == 0, msg)
 
-    @generated_test
+    @acts.signals.generated_test
     def test_passpoint_connect_negative(self):
         """Test connecting to enterprise networks.