diff --git a/hack/boilerplate/boilerplate_test.py b/hack/boilerplate/boilerplate_test.py index 389e812a599..09ea600505d 100644 --- a/hack/boilerplate/boilerplate_test.py +++ b/hack/boilerplate/boilerplate_test.py @@ -20,33 +20,33 @@ from io import StringIO import os import sys + class TestBoilerplate(unittest.TestCase): - """ - Note: run this test from the hack/boilerplate directory. + """ + Note: run this test from the hack/boilerplate directory. - $ python -m unittest boilerplate_test - """ + $ python -m unittest boilerplate_test + """ - def test_boilerplate(self): - os.chdir("test/") + def test_boilerplate(self): + os.chdir("test/") - class Args(object): - def __init__(self): - self.filenames = [] - self.rootdir = "." - self.boilerplate_dir = "../" - self.verbose = True + class Args(object): + def __init__(self): + self.filenames = [] + self.rootdir = "." + self.boilerplate_dir = "../" + self.verbose = True - # capture stdout - old_stdout = sys.stdout - sys.stdout = StringIO.StringIO() + # capture stdout + old_stdout = sys.stdout + sys.stdout = StringIO.StringIO() - boilerplate.args = Args() - ret = boilerplate.main() + boilerplate.args = Args() + ret = boilerplate.main() - output = sorted(sys.stdout.getvalue().split()) + output = sorted(sys.stdout.getvalue().split()) - sys.stdout = old_stdout + sys.stdout = old_stdout - self.assertEqual( - output, ['././fail.go', '././fail.py']) + self.assertEqual(output, ["././fail.go", "././fail.py"])