| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright 2014 The Chromium Authors. All rights reserved. | 2 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 import logging | 6 import logging |
| 7 import memory_inspector | 7 import os |
| 8 import sys | 8 import sys |
| 9 import unittest | 9 import unittest |
| 10 | 10 |
| 11 | 11 |
| 12 if __name__ == '__main__': | 12 if __name__ == '__main__': |
| 13 logging.basicConfig( | 13 logging.basicConfig( |
| 14 level=logging.DEBUG if '-v' in sys.argv else logging.WARNING, | 14 level=logging.DEBUG if '-v' in sys.argv else logging.WARNING, |
| 15 format='%(levelname)5s %(filename)15s(%(lineno)3d): %(message)s') | 15 format='%(levelname)5s %(filename)15s(%(lineno)3d): %(message)s') |
| 16 | 16 |
| 17 suite = unittest.TestSuite() | 17 suite = unittest.TestSuite() |
| 18 loader = unittest.TestLoader() | 18 loader = unittest.TestLoader() |
| 19 suite.addTests(loader.discover(start_dir=memory_inspector.ROOT_DIR, | 19 suite.addTests(loader.discover(start_dir=os.path.dirname(__file__), |
| 20 pattern='*_unittest.py')) | 20 pattern='*_unittest.py')) |
| 21 res = unittest.TextTestRunner(verbosity=2).run(suite) | 21 res = unittest.TextTestRunner(verbosity=2).run(suite) |
| 22 if res.wasSuccessful(): | 22 if res.wasSuccessful(): |
| 23 sys.exit(0) | 23 sys.exit(0) |
| 24 else: | 24 else: |
| 25 sys.exit(1) | 25 sys.exit(1) |
| OLD | NEW |