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)