29 lines
587 B
Python
29 lines
587 B
Python
#! /usr/bin/env python
|
|
import static_pdfs
|
|
|
|
from pdfrw import PdfReader
|
|
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
|
|
|
|
class TestPdfReaderInit(unittest.TestCase):
|
|
|
|
def test_fname_binary_filelike(self):
|
|
with open(static_pdfs.pdffiles[0][0], 'rb') as pdf_file:
|
|
PdfReader(pdf_file)
|
|
|
|
def test_fdata_binary(self):
|
|
with open(static_pdfs.pdffiles[0][0], 'rb') as pdf_file:
|
|
pdf_bytes = pdf_file.read()
|
|
PdfReader(fdata=pdf_bytes)
|
|
|
|
|
|
def main():
|
|
unittest.main()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|