Index: tools/perf/measurements/screenshot_unittest.py |
diff --git a/tools/perf/measurements/screenshot_unittest.py b/tools/perf/measurements/screenshot_unittest.py |
index 0addf1cd76c9de800d095b53468290b5a44d6b42..27a61322852beb68e6bb3e091dc1452344ed62cd 100644 |
--- a/tools/perf/measurements/screenshot_unittest.py |
+++ b/tools/perf/measurements/screenshot_unittest.py |
@@ -8,13 +8,12 @@ import tempfile |
from measurements import screenshot |
from telemetry import benchmark |
-from telemetry.page import page_measurement_unittest_base |
from telemetry.page import page_test |
+from telemetry.testing import page_test_test_case |
from telemetry.unittest import options_for_unittests |
-class ScreenshotUnitTest( |
- page_measurement_unittest_base.PageMeasurementUnitTestBase): |
+class ScreenshotUnitTest(page_test_test_case.PageTestTestCase): |
def setUp(self): |
self._options = options_for_unittests.GetCopy() |
self._options.png_outdir = tempfile.mkdtemp('_png_test') |