def test_to_json(self, mock_normalize):
results = mock.MagicMock()
results.to_dict.return_value = {"a": 1, "b": 20}
mock_normalize.return_value = {"c": 10}
self.assertEqual(json.dumps(mock_normalize.return_value, indent=2),
reports.to_json(results))
mock_normalize.assert_called_once_with(results.to_dict.return_value)
def test_to_json(self, mock_normalize):
results = mock.MagicMock()
results.to_dict.return_value = {"a": 1, "b": 20}
mock_normalize.return_value = {"c": 10}
self.assertEqual(json.dumps(mock_normalize.return_value, indent=2),
reports.to_json(results))
mock_normalize.assert_called_once_with(results.to_dict.return_value)
def main():
if len(sys.argv) == 1:
print_help()
elif len(sys.argv) == 2:
report = reports.to_json(trace_module(sys.argv[1]))
sys.stdout.write(report)
elif len(sys.argv) == 3:
if "--html" in sys.argv[1:]:
arg_pos = 1 if sys.argv[1] != "--html" else 2
report = reports.to_html(trace_module(sys.argv[arg_pos]))
sys.stdout.write(report)
else:
print_help()
raise SystemExit("Wrong input arguments: %s" % sys.argv)
else:
print_help()
raise SystemExit("Wrong input arguments: %s" % sys.argv)
def main():
if len(sys.argv) == 1:
print_help()
elif len(sys.argv) == 2:
report = reports.to_json(trace_module(sys.argv[1]))
sys.stdout.write(report)
elif len(sys.argv) == 3:
if "--html" in sys.argv[1:]:
arg_pos = 1 if sys.argv[1] != "--html" else 2
report = reports.to_html(trace_module(sys.argv[arg_pos]))
sys.stdout.write(report)
else:
print_help()
raise SystemExit("Wrong input arguments: %s" % sys.argv)
else:
print_help()
raise SystemExit("Wrong input arguments: %s" % sys.argv)
