Index: build/android/devil/android/device_temp_file_test.py |
diff --git a/build/android/devil/android/device_temp_file_test.py b/build/android/devil/android/device_temp_file_test.py |
index 76f9cc8ec0179984831eff7fd3028299487fc3e3..6963456f92989b2899f8fcd9077986fcd34fb348 100755 |
--- a/build/android/devil/android/device_temp_file_test.py |
+++ b/build/android/devil/android/device_temp_file_test.py |
@@ -8,18 +8,17 @@ Unit tests for the contents of device_temp_file.py. |
""" |
import logging |
-import os |
import sys |
import unittest |
+from devil import devil_env |
from devil.android import device_errors |
from devil.android import device_temp_file |
from devil.android.sdk import adb_wrapper |
from devil.utils import mock_calls |
-from pylib import constants |
-sys.path.append(os.path.join( |
- constants.DIR_SOURCE_ROOT, 'third_party', 'pymock')) |
+devil_env.InitializeForTests() |
+sys.path.append(devil_env.instance.pymock_path) |
import mock # pylint: disable=F0401 |
class DeviceTempFileTest(mock_calls.TestCase): |