collective/mediawiki_dump_tools
18
0
Fork 0
Code Issues Pull Requests Activity

Merge branch 'parquet_support' into test-parquet

Browse Source
This commit is contained in:
Will Beason
2025-05-29 10:21:30 -05:00
parent ee01ce3e61 ab280dd765
commit 9009bb6fa4
5 changed files with 53 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
test/Wikiq_Unit_Test.py
View File

@@ -1,4 +1,3 @@
import math
import unittest
import os
import subprocess
@@ -7,12 +6,10 @@ from shutil import copyfile
import numpy as np
import pandas as pd
from pandas import DataFrame
from pandas._testing import assert_series_equal
from pandas.testing import assert_frame_equal
from pandas.testing import assert_frame_equal, assert_series_equal
from io import StringIO
import tracemalloc
from typing import Final
from datetime import datetime
# Make references to files and wikiq relative to this file, not to the current working directory.
TEST_DIR: Final[str] = os.path.dirname(os.path.realpath(__file__))
@@ -181,7 +178,7 @@ class WikiqTestCase(unittest.TestCase):
tester = WikiqTester(SAILORMOON, "collapse-user", in_compression="7z")
try:
tester.call_wikiq("--collapse-user")
tester.call_wikiq("--collapse-user", "--fandom-2020")
except subprocess.CalledProcessError as exc:
self.fail(exc.stderr.decode("utf8"))
@@ -195,7 +192,7 @@ class WikiqTestCase(unittest.TestCase):
tester = WikiqTester(SAILORMOON, "persistence_segment", in_compression="7z")
try:
tester.call_wikiq("--persistence segment")
tester.call_wikiq("--persistence segment", "--fandom-2020")
except subprocess.CalledProcessError as exc:
self.fail(exc.stderr.decode("utf8"))
@@ -209,7 +206,7 @@ class WikiqTestCase(unittest.TestCase):
tester = WikiqTester(SAILORMOON, "persistence_legacy", in_compression="7z")
try:
tester.call_wikiq("--persistence legacy")
tester.call_wikiq("--persistence legacy", "--fandom-2020")
except subprocess.CalledProcessError as exc:
self.fail(exc.stderr.decode("utf8"))
@@ -223,7 +220,7 @@ class WikiqTestCase(unittest.TestCase):
tester = WikiqTester(SAILORMOON, "persistence", in_compression="7z")
try:
tester.call_wikiq("--persistence")
tester.call_wikiq("--persistence", "--fandom-2020")
except subprocess.CalledProcessError as exc:
self.fail(exc.stderr.decode("utf8"))
@@ -239,7 +236,7 @@ class WikiqTestCase(unittest.TestCase):
tester = WikiqTester(SAILORMOON, "url-encode", in_compression="7z")
try:
tester.call_wikiq("--url-encode")
tester.call_wikiq("--url-encode", "--fandom-2020")
except subprocess.CalledProcessError as exc:
self.fail(exc.stderr.decode("utf8"))
@@ -266,7 +263,7 @@ class WikiqTestCase(unittest.TestCase):
tester = WikiqTester(wiki=SAILORMOON, case_name="noargs", in_compression="7z")
try:
outs = tester.call_wikiq("--stdout", out=False).decode("utf8")
outs = tester.call_wikiq( "--stdout", "--fandom-2020", out=False).decode("utf8")
except subprocess.CalledProcessError as exc:
self.fail(exc.stderr.decode("utf8"))