fix: soften 818 constraints

bigcode-project · May 1, 2024 · 7efa852 · 7efa852
1 parent b58494d
commit 7efa852
Show file tree

Hide file tree

Showing 5 changed files with 637 additions and 637 deletions.
diff --git a/data/clean/f_818_wenhao.py b/data/clean/f_818_wenhao.py
@@ -61,7 +61,7 @@ def test_case_2(self):
         df = f_818(5, columns=custom_columns, seed=0)
         self.assertTrue(all(column in custom_columns for column in df.columns))
         # assert first 2 rows data
-        self.assertEqual(df.iloc[0].tolist(), [0.5488135039273248, 0.7151893663724195, 0.6027633760716439])
+        self.assertEqual(set(df.iloc[0].tolist()), {0.5488135039273248, 0.7151893663724195, 0.6027633760716439})
 
     def test_case_3(self):
         # Test custom rows
@@ -71,7 +71,7 @@ def test_case_3(self):
 
     def test_case_4(self):
         df = f_818(5, seed=42)
-        self.assertEqual(df.iloc[0].tolist(), [0.3745401188473625, 0.9507143064099162, 0.7319939418114051, 0.5986584841970366, 0.15601864044243652])
+        self.assertEqual(set(df.iloc[0].tolist()), {0.3745401188473625, 0.9507143064099162, 0.7319939418114051, 0.5986584841970366, 0.15601864044243652})
 
     def test_case_5(self):
         # Test handling edge cases - negative rows