diff options
Diffstat (limited to 'afdo_redaction/remove_indirect_calls_test.py')
-rwxr-xr-x | afdo_redaction/remove_indirect_calls_test.py | 33 |
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() |