diff --git a/test/Wikiq_Unit_Test.py b/test/Wikiq_Unit_Test.py index df33905..8191860 100644 --- a/test/Wikiq_Unit_Test.py +++ b/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))