From 0468799a56d7fed0df6984fb89349a9cd2c35e2b Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Thu, 23 Jul 2015 07:36:28 -0400 Subject: Test (and fix!) the secret data dumper. --- coverage/data.py | 24 +++++++----------------- 1 file changed, 7 insertions(+), 17 deletions(-) (limited to 'coverage/data.py') 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:]) -- cgit v1.2.1