From c0e6979a98251fa2300b2c61f206775ef953f7a8 Mon Sep 17 00:00:00 2001 From: Jeremy Singer-Vine Date: Sun, 6 Mar 2016 11:41:48 -0500 Subject: [PATCH] Add test for very basic things --- tests/test-basics.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 tests/test-basics.py diff --git a/tests/test-basics.py b/tests/test-basics.py new file mode 100644 index 0000000..a32930a --- /dev/null +++ b/tests/test-basics.py @@ -0,0 +1,24 @@ +#!/usr/bin/env python +import unittest +import pandas as pd +import pdfplumber +import sys, os +import six + +import logging +logging.disable(logging.ERROR) + +HERE = os.path.abspath(os.path.dirname(__file__)) + +class Test(unittest.TestCase): + + def setUp(self): + path = os.path.join(HERE, "pdfs/nics-background-checks-2015-11.pdf") + self.pdf = pdfplumber.from_path(path) + + def test_metadata(self): + metadata = self.pdf.metadata + assert(isinstance(metadata["Producer"], six.text_type)) + + def test_pagecount(self): + assert(len(self.pdf.pages) == 1)