diff --git a/test/test_report.py b/test/test_report.py index 7f309d3f..691eacd9 100644 --- a/test/test_report.py +++ b/test/test_report.py @@ -51,9 +51,7 @@ def test_print_report(self): # Print the report response = build.printReport(template) - for key in ['pk', 'model_type', 'output', 'template']: + for key in ['pk', 'output']: self.assertIn(key, response) self.assertIsNotNone(response['output']) - self.assertEqual(response['template'], template.pk) - self.assertEqual(response['model_type'], build.getModelType())