Skip to content

Commit a06d483

Browse files
414owentorymur
authored andcommitted
Stabilize python guide tests
This change was cherry-picked from #165 It's separate and quite important, as it's blocking everything else...
1 parent b2c419a commit a06d483

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/test_guide.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -73,9 +73,9 @@ def test_token_trans_keys_identical():
7373
guide1 = Guide(index)
7474
guide2 = Guide(index)
7575

76-
assert guide1.advance(3) == guide2.advance(3)
76+
assert sorted(guide1.advance(3)) == sorted(guide2.advance(3))
7777
# `a` and `b` have similar transitions to `z`
78-
assert guide1.advance(1) == guide2.advance(2)
78+
assert sorted(guide1.advance(1)) == sorted(guide2.advance(2))
7979
assert guide1.advance(3) == guide2.advance(3) == [eos_token_id]
8080
assert guide1.is_finished()
8181
assert guide2.is_finished()
@@ -94,9 +94,9 @@ def test_str_and_bytes_produce_the_same():
9494
guide1 = Guide(index1)
9595
guide2 = Guide(index2)
9696

97-
assert guide1.advance(3) == guide2.advance(3)
97+
assert sorted(guide1.advance(3)) == sorted(guide2.advance(3))
9898
# `a` and `b` have similar transitions to `z`
99-
assert guide1.advance(1) == guide2.advance(2)
99+
assert sorted(guide1.advance(1)) == sorted(guide2.advance(2))
100100
assert guide1.advance(3) == guide2.advance(3) == [eos_token_id]
101101
assert guide1.is_finished()
102102
assert guide2.is_finished()

0 commit comments

Comments
 (0)