fix tests.

test/Wikiq_Unit_Test.py

@@ -129,7 +129,7 @@ def test_WP_namespaces():
     # as a test let's make sure that we get equal data frames
     test = pd.read_table(tester.output)
     num_wrong_ns = sum(~test.namespace.isin({0, 1}))
-    self.assertEqual(num_wrong_ns, 0)
+    assert num_wrong_ns == 0
     baseline = pd.read_table(tester.baseline_file)
     assert_frame_equal(test, baseline, check_like=True)
 
@@ -144,7 +144,7 @@ def test_WP_revert_radius():
     # as a test let's make sure that we get equal data frames
     test = pd.read_table(tester.output)
     num_wrong_ns = sum(~test.namespace.isin({0, 1}))
-    self.assertEqual(num_wrong_ns, 0)
+    assert num_wrong_ns == 0
     baseline = pd.read_table(tester.baseline_file)
     assert_frame_equal(test, baseline, check_like=True)
 
@@ -159,7 +159,7 @@ def test_WP_no_revert_radius():
     # as a test let's make sure that we get equal data frames
     test = pd.read_table(tester.output)
     num_reverted = sum(i is None for i in test.revert)
-    self.assertEqual(num_reverted, 0)
+    assert num_reverted == 0
     baseline = pd.read_table(tester.baseline_file)
     assert_frame_equal(test, baseline, check_like=True)
 
@@ -207,8 +207,8 @@ def test_pwr_wikidiff2():
     except subprocess.CalledProcessError as exc:
         pytest.fail(exc.stderr.decode("utf8"))
 
-    test = pd.read_parquet(tester.output)
-    baseline = pd.read_parquet(tester.baseline_file)
+    test = pd.read_table(tester.output)
+    baseline = pd.read_table(tester.baseline_file)
     assert_frame_equal(test, baseline, check_like=True)
 
 def test_pwr_segment():
@@ -287,7 +287,7 @@ def test_malformed_noargs():
         tester.call_wikiq()
     except subprocess.CalledProcessError as exc:
         errlines = exc.stderr.decode("utf8").splitlines()
-        self.assertEqual(errlines[-1], want_exception)
+        assert errlines[-1] == want_exception
     else:
         pytest.fail("No exception raised, want: {}".format(want_exception))