X-Git-Url: https://code.communitydata.science/mediawiki_dump_tools.git/blobdiff_plain/dba793c6ac595e7a5c0ac795575c28231f06f8cb..afd40c1a45166602773aefe35e10a202e8dc47dd:/test/Wikiq_Unit_Test.py diff --git a/test/Wikiq_Unit_Test.py b/test/Wikiq_Unit_Test.py index e43ee7d..0826f1f 100644 --- a/test/Wikiq_Unit_Test.py +++ b/test/Wikiq_Unit_Test.py @@ -108,6 +108,26 @@ class Test_Basic(unittest.TestCase): baseline = pd.read_table(baseline_file) assert_frame_equal(test,baseline) + def test_pwr_segment(self): + test_filename = "persistence_segment_" + self.wikiq_out_name + test_file = os.path.join(self.test_output_dir, test_filename) + if os.path.exists(test_file): + os.remove(test_file) + + call = self.base_call.format(self.input_file, self.test_output_dir) + call = call + " --persistence segment" + proc = subprocess.Popen(call,stdout=subprocess.PIPE,shell=True) + proc.wait() + + + copyfile(self.call_output, test_file) + + baseline_file = os.path.join(".", self.baseline_output_dir, test_filename) + + test = pd.read_table(test_file) + baseline = pd.read_table(baseline_file) + assert_frame_equal(test,baseline) + def test_pwr_legacy(self): test_filename = "persistence_legacy_" + self.wikiq_out_name test_file = os.path.join(self.test_output_dir, test_filename) @@ -115,7 +135,7 @@ class Test_Basic(unittest.TestCase): os.remove(test_file) call = self.base_call.format(self.input_file, self.test_output_dir) - call = call + " --persistence-legacy" + call = call + " --persistence legacy" proc = subprocess.Popen(call,stdout=subprocess.PIPE,shell=True) proc.wait() @@ -131,8 +151,8 @@ class Test_Basic(unittest.TestCase): def test_pwr(self): test_filename = "persistence_" + self.wikiq_out_name test_file = os.path.join(self.test_output_dir, test_filename) - if os.path.exists(test_file): - os.remove(test_file) + if os.path.exists(test_file): + os.remove(test_file) call = self.base_call.format(self.input_file, self.test_output_dir) call = call + " --persistence"