Move error tests from test_common to test_errors

Issue atlanhq#162
rbares · Oct 28, 2018 · 555c4ae · 555c4ae
1 parent f5ff086
commit 555c4ae
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 13 deletions.
diff --git a/tests/test_common.py b/tests/test_common.py
@@ -37,19 +37,6 @@ def test_password():
 
  filename = os.path.join(testdir, "health_protected.pdf")
 
- try:
- tables = camelot.read_pdf(filename, flavor="stream")
- assert false
- except Exception as err:
- assert "file has not been decrypted" in str(err)
-
- try:
- tables = camelot.read_pdf(filename, flavor="stream",
- password="wrongpass")
- assert false
- except Exception as err:
- assert "file has not been decrypted" in str(err)
-
  tables = camelot.read_pdf(filename, flavor="stream",
  password="ownerpass")
  assert df.equals(tables[0].df)

diff --git a/tests/test_errors.py b/tests/test_errors.py
@@ -75,3 +75,17 @@ def _find_executable_patch(arg):
  filename = os.path.join(testdir, 'foo.pdf')
  with pytest.raises(Exception, message=message):
  tables = camelot.read_pdf(filename)
+
+
+def test_no_password():
+ filename = os.path.join(testdir, 'health_protected.pdf')
+ message = 'file has not been decrypted'
+ with pytest.raises(Exception, message=message):
+ tables = camelot.read_pdf(filename)
+
+
+def test_bad_password():
+ filename = os.path.join(testdir, 'health_protected.pdf')
+ message = 'file has not been decrypted'
+ with pytest.raises(Exception, message=message):
+ tables = camelot.read_pdf(filename, password='wrongpass')