Python BareResourceGenerator.pdf Examples

Example #1

File: test_BaseResourceGenerator.py Project: was-dtech/cs-unplugged

 def test_pdf_multiple_pages(self):
     generator = BareResourceGenerator()
     generator.data = MagicMock(
         return_value=[
             {"type": "html", "data": "Page 1"},
             {"type": "html", "data": "Page 2"},
         ]
     )
     (pdf_file, filename) = generator.pdf("Test")
     pdf = PdfFileReader(BytesIO(pdf_file))
     self.assertEqual(pdf.getNumPages(), 2)

Example #2

File: test_BaseResourceGenerator.py Project: was-dtech/cs-unplugged

 def test_pdf_single_page(self):
     generator = BareResourceGenerator()
     (pdf_file, filename) = generator.pdf("Test")
     pdf = PdfFileReader(BytesIO(pdf_file))
     self.assertEqual(pdf.getNumPages(), 1)