aboutsummaryrefslogtreecommitdiff
path: root/heatmaps/heatmap_generator_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'heatmaps/heatmap_generator_test.py')
-rwxr-xr-xheatmaps/heatmap_generator_test.py9
1 files changed, 5 insertions, 4 deletions
diff --git a/heatmaps/heatmap_generator_test.py b/heatmaps/heatmap_generator_test.py
index 0c0bbfc8..5008c653 100755
--- a/heatmaps/heatmap_generator_test.py
+++ b/heatmaps/heatmap_generator_test.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python2
+#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2018 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
@@ -8,11 +8,12 @@
from __future__ import division, print_function
-import mock
-import os
+import unittest.mock as mock
import unittest
-import heatmap_generator
+import os
+
+from heatmaps import heatmap_generator
def _write_perf_mmap(pid, tid, addr, size, fp):