diff options
Diffstat (limited to 'coverage/data.py')
-rw-r--r-- | coverage/data.py | 24 |
1 files changed, 7 insertions, 17 deletions
diff --git a/coverage/data.py b/coverage/data.py index b0d6838..77bfd38 100644 --- a/coverage/data.py +++ b/coverage/data.py @@ -8,7 +8,7 @@ import socket from coverage import env from coverage.backward import iitems, string_class -from coverage.debug import _TEST_NAME_FILE +from coverage.debug import _TEST_NAME_FILE, pretty_data from coverage.files import PathAliases from coverage.misc import CoverageException, file_be_gone @@ -542,7 +542,7 @@ class CoverageDataFiles(object): os.remove(f) -def debug_main(): +def debug_main(args): """Dump the raw data from data files. Run this as:: @@ -550,22 +550,12 @@ def debug_main(): $ python -m coverage.data [FILE] """ - from coverage.debug import pretty_data - import sys - - if len(sys.argv) > 1: - files = sys.argv[1:] - else: - files = [".coverage"] - - for filename in files: - data = CoverageData._read_raw_data_file(filename) + for filename in (args or [".coverage"]): print("--- {0} ------------------------------".format(filename)) - if data: - print(pretty_data(data)) - else: - print("No data collected") + data = CoverageData._read_raw_data_file(filename) + print(pretty_data(data)) if __name__ == '__main__': - debug_main() + import sys + debug_main(sys.argv[1:]) |