def _test_extract_document_mangled(self, testname, resultname):
source = read_file("onecaseperfile", testname)
expected_result = read_file("onecaseperfile", resultname)
doc_dict = pysource.extract_documents(source)
self.assertEqual(len(doc_dict), 1)
doc = doc_dict[None]
self.assertEqual(doc.recreate(), expected_result)
self.assertTrue(doc.has_errors())
def _test_extract_document_noerrors(self, testname):
source = read_file("onecaseperfile", testname)
expected_result = read_file("onecaseperfile", "rst")
doc_dict = pysource.extract_documents(source)
self.assertEqual(len(doc_dict), 1)
doc = doc_dict[None]
self.assertEqual(list(doc.errors()), [])
self.assertEqual(doc.recreate(), expected_result)
self.assertEqual(list(doc.errors_lastrecreate()), [])
self.assertFalse(doc.has_errors())
def _test_extract_documents_noerrors(self, doc_num, testname):
source = read_file("multiplecasesperfile", testname)
doc_dict = pysource.extract_documents(source)
self.assertEqual(len(doc_dict), doc_num)
for doc_id, doc in doc_dict.items():
if doc_id is None:
doc_id = "none"
filename = "{}.rst".format(doc_id)
expected_result = read_file("multiplecasesperfile", filename)
self.assertEqual(list(doc.errors()), [])
self.assertEqual(doc.recreate(), expected_result)
self.assertEqual(list(doc.errors_lastrecreate()), [])
self.assertFalse(doc.has_errors())
def test_extract_document_null(self):
source = read_file("onecaseperfile", "null.py")
doc_dict = pysource.extract_documents(source)
self.assertEqual(doc_dict, {})
