fix: make 270 more robust

Author: terryyz

data/clean/f_270_haolan_ratna_edit.py

@@ -56,20 +56,21 @@ def f_270(output_file, test_directory):
 from unittest.mock import patch, MagicMock
 from collections import Counter
 from faker import Faker
-
+import shutil
 
 
 # Blackbox test cases
 class TestCases(unittest.TestCase):
     def setUp(self):
         self.test_directory = './testdir_f270'
+        os.makedirs(self.test_directory, exist_ok=True)
+        
         self.output_file = 'test_output.csv'
         self.list_files = []
 
     # Function to create fake dictionary files
     def create_fake_dict_files(self, directory, num_files, num_words):
         fake = Faker()
-        os.makedirs(directory, exist_ok=True)
         for _ in range(num_files):
             file_name = fake.file_name(extension='txt')
             self.list_files.append(os.path.join(directory, file_name))
@@ -89,15 +90,15 @@ def tearDown(self):
         if os.path.exists('test_output.csv'):
             os.remove('test_output.csv')
         if os.path.exists(self.test_directory):
-            os.rmdir(self.test_directory)
+            shutil.rmtree(self.test_directory)
 
     def test_no_files_in_directory(self):
         # Test case where there are no txt files in the directory
         self.create_fake_dict_files(self.test_directory, 0, 0)
         result = f_270(self.output_file, self.test_directory)
         self.assertEqual(result, 0)
         self.remove_files()
-
+    
     def test_single_file_multiple_words(self):
         # Test case with a single file containing multiple words
         self.create_fake_dict_files(self.test_directory, 1, 50)
@@ -112,7 +113,6 @@ def test_multiple_files_multiple_words(self):
         result = f_270(self.output_file, self.test_directory)
         self.remove_files()
         self.assertEqual(100,result)
-        # self.assertFalse(result)
 
     def test_directory_does_not_exist(self):
         # Test case where the specified directory does not exist

data/processed/287_wo_doc.py

@@ -53,16 +53,18 @@ def task_func(output_file, test_directory):
 from unittest.mock import patch, MagicMock
 from collections import Counter
 from faker import Faker
+import shutil
 # Blackbox test cases
 class TestCases(unittest.TestCase):
     def setUp(self):
         self.test_directory = './testdir_f270'
+        os.makedirs(self.test_directory, exist_ok=True)
+        
         self.output_file = 'test_output.csv'
         self.list_files = []
     # Function to create fake dictionary files
     def create_fake_dict_files(self, directory, num_files, num_words):
         fake = Faker()
-        os.makedirs(directory, exist_ok=True)
         for _ in range(num_files):
             file_name = fake.file_name(extension='txt')
             self.list_files.append(os.path.join(directory, file_name))
@@ -81,13 +83,14 @@ def tearDown(self):
         if os.path.exists('test_output.csv'):
             os.remove('test_output.csv')
         if os.path.exists(self.test_directory):
-            os.rmdir(self.test_directory)
+            shutil.rmtree(self.test_directory)
     def test_no_files_in_directory(self):
         # Test case where there are no txt files in the directory
         self.create_fake_dict_files(self.test_directory, 0, 0)
         result = task_func(self.output_file, self.test_directory)
         self.assertEqual(result, 0)
         self.remove_files()
+    
     def test_single_file_multiple_words(self):
         # Test case with a single file containing multiple words
         self.create_fake_dict_files(self.test_directory, 1, 50)
@@ -100,7 +103,6 @@ def test_multiple_files_multiple_words(self):
         result = task_func(self.output_file, self.test_directory)
         self.remove_files()
         self.assertEqual(100,result)
-        # self.assertFalse(result)
     def test_directory_does_not_exist(self):
         # Test case where the specified directory does not exist
         result = task_func(self.output_file, self.test_directory)

data/raw/f_270_haolan_ratna_edit.py

@@ -56,20 +56,21 @@ def f_270(output_file, test_directory):
 from unittest.mock import patch, MagicMock
 from collections import Counter
 from faker import Faker
-
+import shutil
 
 
 # Blackbox test cases
 class TestCases(unittest.TestCase):
     def setUp(self):
         self.test_directory = './testdir_f270'
+        os.makedirs(self.test_directory, exist_ok=True)
+        
         self.output_file = 'test_output.csv'
         self.list_files = []
 
     # Function to create fake dictionary files
     def create_fake_dict_files(self, directory, num_files, num_words):
         fake = Faker()
-        os.makedirs(directory, exist_ok=True)
         for _ in range(num_files):
             file_name = fake.file_name(extension='txt')
             self.list_files.append(os.path.join(directory, file_name))
@@ -89,15 +90,15 @@ def tearDown(self):
         if os.path.exists('test_output.csv'):
             os.remove('test_output.csv')
         if os.path.exists(self.test_directory):
-            os.rmdir(self.test_directory)
+            shutil.rmtree(self.test_directory)
 
     def test_no_files_in_directory(self):
         # Test case where there are no txt files in the directory
         self.create_fake_dict_files(self.test_directory, 0, 0)
         result = f_270(self.output_file, self.test_directory)
         self.assertEqual(result, 0)
         self.remove_files()
-
+    
     def test_single_file_multiple_words(self):
         # Test case with a single file containing multiple words
         self.create_fake_dict_files(self.test_directory, 1, 50)
@@ -112,7 +113,6 @@ def test_multiple_files_multiple_words(self):
         result = f_270(self.output_file, self.test_directory)
         self.remove_files()
         self.assertEqual(100,result)
-        # self.assertFalse(result)
 
     def test_directory_does_not_exist(self):
         # Test case where the specified directory does not exist