Skip to content

Commit 809f9e6

Browse files
committed
fix: make 270 more robust
1 parent 2e256c9 commit 809f9e6

File tree

5 files changed

+1117
-1115
lines changed

5 files changed

+1117
-1115
lines changed

data/clean/f_270_haolan_ratna_edit.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -56,20 +56,21 @@ def f_270(output_file, test_directory):
5656
from unittest.mock import patch, MagicMock
5757
from collections import Counter
5858
from faker import Faker
59-
59+
import shutil
6060

6161

6262
# Blackbox test cases
6363
class TestCases(unittest.TestCase):
6464
def setUp(self):
6565
self.test_directory = './testdir_f270'
66+
os.makedirs(self.test_directory, exist_ok=True)
67+
6668
self.output_file = 'test_output.csv'
6769
self.list_files = []
6870

6971
# Function to create fake dictionary files
7072
def create_fake_dict_files(self, directory, num_files, num_words):
7173
fake = Faker()
72-
os.makedirs(directory, exist_ok=True)
7374
for _ in range(num_files):
7475
file_name = fake.file_name(extension='txt')
7576
self.list_files.append(os.path.join(directory, file_name))
@@ -89,15 +90,15 @@ def tearDown(self):
8990
if os.path.exists('test_output.csv'):
9091
os.remove('test_output.csv')
9192
if os.path.exists(self.test_directory):
92-
os.rmdir(self.test_directory)
93+
shutil.rmtree(self.test_directory)
9394

9495
def test_no_files_in_directory(self):
9596
# Test case where there are no txt files in the directory
9697
self.create_fake_dict_files(self.test_directory, 0, 0)
9798
result = f_270(self.output_file, self.test_directory)
9899
self.assertEqual(result, 0)
99100
self.remove_files()
100-
101+
101102
def test_single_file_multiple_words(self):
102103
# Test case with a single file containing multiple words
103104
self.create_fake_dict_files(self.test_directory, 1, 50)
@@ -112,7 +113,6 @@ def test_multiple_files_multiple_words(self):
112113
result = f_270(self.output_file, self.test_directory)
113114
self.remove_files()
114115
self.assertEqual(100,result)
115-
# self.assertFalse(result)
116116

117117
def test_directory_does_not_exist(self):
118118
# Test case where the specified directory does not exist

data/processed/287_wo_doc.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -53,16 +53,18 @@ def task_func(output_file, test_directory):
5353
from unittest.mock import patch, MagicMock
5454
from collections import Counter
5555
from faker import Faker
56+
import shutil
5657
# Blackbox test cases
5758
class TestCases(unittest.TestCase):
5859
def setUp(self):
5960
self.test_directory = './testdir_f270'
61+
os.makedirs(self.test_directory, exist_ok=True)
62+
6063
self.output_file = 'test_output.csv'
6164
self.list_files = []
6265
# Function to create fake dictionary files
6366
def create_fake_dict_files(self, directory, num_files, num_words):
6467
fake = Faker()
65-
os.makedirs(directory, exist_ok=True)
6668
for _ in range(num_files):
6769
file_name = fake.file_name(extension='txt')
6870
self.list_files.append(os.path.join(directory, file_name))
@@ -81,13 +83,14 @@ def tearDown(self):
8183
if os.path.exists('test_output.csv'):
8284
os.remove('test_output.csv')
8385
if os.path.exists(self.test_directory):
84-
os.rmdir(self.test_directory)
86+
shutil.rmtree(self.test_directory)
8587
def test_no_files_in_directory(self):
8688
# Test case where there are no txt files in the directory
8789
self.create_fake_dict_files(self.test_directory, 0, 0)
8890
result = task_func(self.output_file, self.test_directory)
8991
self.assertEqual(result, 0)
9092
self.remove_files()
93+
9194
def test_single_file_multiple_words(self):
9295
# Test case with a single file containing multiple words
9396
self.create_fake_dict_files(self.test_directory, 1, 50)
@@ -100,7 +103,6 @@ def test_multiple_files_multiple_words(self):
100103
result = task_func(self.output_file, self.test_directory)
101104
self.remove_files()
102105
self.assertEqual(100,result)
103-
# self.assertFalse(result)
104106
def test_directory_does_not_exist(self):
105107
# Test case where the specified directory does not exist
106108
result = task_func(self.output_file, self.test_directory)

data/raw/f_270_haolan_ratna_edit.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -56,20 +56,21 @@ def f_270(output_file, test_directory):
5656
from unittest.mock import patch, MagicMock
5757
from collections import Counter
5858
from faker import Faker
59-
59+
import shutil
6060

6161

6262
# Blackbox test cases
6363
class TestCases(unittest.TestCase):
6464
def setUp(self):
6565
self.test_directory = './testdir_f270'
66+
os.makedirs(self.test_directory, exist_ok=True)
67+
6668
self.output_file = 'test_output.csv'
6769
self.list_files = []
6870

6971
# Function to create fake dictionary files
7072
def create_fake_dict_files(self, directory, num_files, num_words):
7173
fake = Faker()
72-
os.makedirs(directory, exist_ok=True)
7374
for _ in range(num_files):
7475
file_name = fake.file_name(extension='txt')
7576
self.list_files.append(os.path.join(directory, file_name))
@@ -89,15 +90,15 @@ def tearDown(self):
8990
if os.path.exists('test_output.csv'):
9091
os.remove('test_output.csv')
9192
if os.path.exists(self.test_directory):
92-
os.rmdir(self.test_directory)
93+
shutil.rmtree(self.test_directory)
9394

9495
def test_no_files_in_directory(self):
9596
# Test case where there are no txt files in the directory
9697
self.create_fake_dict_files(self.test_directory, 0, 0)
9798
result = f_270(self.output_file, self.test_directory)
9899
self.assertEqual(result, 0)
99100
self.remove_files()
100-
101+
101102
def test_single_file_multiple_words(self):
102103
# Test case with a single file containing multiple words
103104
self.create_fake_dict_files(self.test_directory, 1, 50)
@@ -112,7 +113,6 @@ def test_multiple_files_multiple_words(self):
112113
result = f_270(self.output_file, self.test_directory)
113114
self.remove_files()
114115
self.assertEqual(100,result)
115-
# self.assertFalse(result)
116116

117117
def test_directory_does_not_exist(self):
118118
# Test case where the specified directory does not exist

0 commit comments

Comments
 (0)