update: improve wenhao tests

bigcode-project · Apr 19, 2024 · 87ea280 · 87ea280
1 parent 1058e89
commit 87ea280
Show file tree

Hide file tree

Showing 4 changed files with 373 additions and 373 deletions.
diff --git a/data/clean/f_750_wenhao.py b/data/clean/f_750_wenhao.py
@@ -69,7 +69,7 @@ def test_plot_titles(self):
         plots = f_750(self.directory, self.pattern)
         expected_titles = ['sales_data_2022.csv', 'sales_data_2021.csv']
         plot_titles = [plot.get_title() for plot in plots]
-        self.assertEqual(plot_titles, expected_titles, "Plot titles should match the CSV filenames")
+        self.assertEqual(set(plot_titles), set(expected_titles), "Plot titles should match the CSV filenames")
 
     def test_no_files_matched(self):
         plots = f_750(self.directory, r"^no_match_\d{4}.csv")