Skip to content

Commit 687df1e

Browse files
committed
fix: table TEDS bug
1 parent baaa753 commit 687df1e

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

tests/test_teds.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -218,10 +218,8 @@ def test_teds_structure_same_content_different(self):
218218
groundtruth=gt,
219219
table_edit_result=self.valid_table_edit_result
220220
)
221+
assert result.score == 0.7999999999999999
221222

222-
self.assertTrue(result.success)
223-
self.assertGreater(result.score, 0.0)
224-
self.assertLess(result.score, 1.0)
225223

226224

227225
class TestTEDSAdvanced(unittest.TestCase):
@@ -326,10 +324,7 @@ def test_teds_content_similarity(self):
326324
table2,
327325
table_edit_result=self.valid_table_edit_result
328326
)
329-
330-
self.assertTrue(result.success)
331-
self.assertGreater(result.score, 0.0)
332-
self.assertLess(result.score, 1.0)
327+
assert result.score == 0.3999999999999999
333328

334329

335330
class TestStructureTEDS(unittest.TestCase):

0 commit comments

Comments
 (0)