aboutsummaryrefslogtreecommitdiff
path: root/afdo_redaction/remove_indirect_calls_test.py
diff options
context:
space:
mode:
authorSadaf Ebrahimi <sadafebrahimi@google.com>2022-11-11 23:35:38 +0000
committerAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>2022-11-11 23:35:38 +0000
commit40214b48188358a80b7478bfff21d4814dd9177c (patch)
tree77dc031614745bb406dbd90cea9a082a1b5cdd54 /afdo_redaction/remove_indirect_calls_test.py
parenta51582ad9cb50ec284f4718765bd5d31fa0069d4 (diff)
parent584b8e46d146a2bcfeffd64448a2d8e92904168d (diff)
downloadtoolchain-utils-40214b48188358a80b7478bfff21d4814dd9177c.tar.gz
Original change: https://android-review.googlesource.com/c/platform/external/toolchain-utils/+/2292563 Change-Id: Ida212764c332af8ba8ab8aaa62a7d2a007e3314e Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
Diffstat (limited to 'afdo_redaction/remove_indirect_calls_test.py')
-rwxr-xr-xafdo_redaction/remove_indirect_calls_test.py33
1 files changed, 16 insertions, 17 deletions
diff --git a/afdo_redaction/remove_indirect_calls_test.py b/afdo_redaction/remove_indirect_calls_test.py
index 164b284f..640b747f 100755
--- a/afdo_redaction/remove_indirect_calls_test.py
+++ b/afdo_redaction/remove_indirect_calls_test.py
@@ -1,12 +1,11 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
-# Copyright 2019 The Chromium OS Authors. All rights reserved.
+# Copyright 2019 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Tests for remove_indirect_calls"""
-from __future__ import print_function
import io
import unittest
@@ -15,22 +14,22 @@ from afdo_redaction import remove_indirect_calls
def _run_test(input_lines):
- input_buf = io.StringIO('\n'.join(input_lines))
- output_buf = io.StringIO()
- remove_indirect_calls.run(input_buf, output_buf)
- return output_buf.getvalue().splitlines()
+ input_buf = io.StringIO("\n".join(input_lines))
+ output_buf = io.StringIO()
+ remove_indirect_calls.run(input_buf, output_buf)
+ return output_buf.getvalue().splitlines()
class Test(unittest.TestCase):
- """Tests"""
+ """Tests"""
- def test_empty_profile(self):
- self.assertEqual(_run_test([]), [])
+ def test_empty_profile(self):
+ self.assertEqual(_run_test([]), [])
- def test_removal_on_real_world_code(self):
- # These are copied from an actual textual AFDO profile, but the names made
- # lints unhappy due to their length, so I had to be creative.
- profile_lines = """_ZLongSymbolName:52862:1766
+ def test_removal_on_real_world_code(self):
+ # These are copied from an actual textual AFDO profile, but the names made
+ # lints unhappy due to their length, so I had to be creative.
+ profile_lines = """_ZLongSymbolName:52862:1766
14: 2483
8.1: _SomeInlinedSym:45413
11: _AndAnother:35481
@@ -45,7 +44,7 @@ class Test(unittest.TestCase):
0: 2483
""".strip().splitlines()
- expected_lines = """_ZLongSymbolName:52862:1766
+ expected_lines = """_ZLongSymbolName:52862:1766
14: 2483
8.1: _SomeInlinedSym:45413
11: _AndAnother:35481
@@ -60,8 +59,8 @@ class Test(unittest.TestCase):
0: 2483
""".strip().splitlines()
- self.assertEqual(_run_test(profile_lines), expected_lines)
+ self.assertEqual(_run_test(profile_lines), expected_lines)
-if __name__ == '__main__':
- unittest.main()
+if __name__ == "__main__":
+ unittest.main()