58 lines
1.9 KiB
Diff
58 lines
1.9 KiB
Diff
From f5a777572fb9c91fa10be3c1ffd5f65cc9323653 Mon Sep 17 00:00:00 2001
|
|
From: fortable1999 <fortable1999@gmail.com>
|
|
Date: Wed, 12 Dec 2018 11:18:50 +0900
|
|
Subject: [PATCH] Fixed test case for ListCorrector; Fixed FSA __eq__ function
|
|
|
|
---
|
|
src/whoosh/automata/fsa.py | 7 +------
|
|
tests/test_reading.py | 1 +
|
|
tests/test_spelling.py | 7 ++++++-
|
|
5 files changed, 12 insertions(+), 8 deletions(-)
|
|
create mode 100644 .gitignore
|
|
|
|
diff --git a/src/whoosh/automata/fsa.py b/src/whoosh/automata/fsa.py
|
|
index 280ddb50..54d23f2f 100644
|
|
--- a/src/whoosh/automata/fsa.py
|
|
+++ b/src/whoosh/automata/fsa.py
|
|
@@ -44,12 +44,7 @@ def __eq__(self, other):
|
|
return False
|
|
st = self.transitions
|
|
ot = other.transitions
|
|
- if list(st) != list(ot):
|
|
- return False
|
|
- for key in st:
|
|
- if st[key] != ot[key]:
|
|
- return False
|
|
- return True
|
|
+ return st == ot
|
|
|
|
def all_states(self):
|
|
stateset = set(self.transitions)
|
|
diff --git a/tests/test_reading.py b/tests/test_reading.py
|
|
index e2c2b71d..9fefc41c 100644
|
|
--- a/tests/test_reading.py
|
|
+++ b/tests/test_reading.py
|
|
@@ -1,3 +1,4 @@
|
|
+# coding=utf-8
|
|
from __future__ import with_statement
|
|
import random, threading, time
|
|
|
|
diff --git a/tests/test_spelling.py b/tests/test_spelling.py
|
|
index ce5284f1..440c2d02 100644
|
|
--- a/tests/test_spelling.py
|
|
+++ b/tests/test_spelling.py
|
|
@@ -20,7 +20,12 @@ def test_list_corrector():
|
|
corr = spelling.ListCorrector(_wordlist)
|
|
typo = "reoction"
|
|
sugs = list(corr.suggest(typo, maxdist=2))
|
|
- target = [w for w in _wordlist if levenshtein(typo, w) <= 2]
|
|
+ target = []
|
|
+ for lev_dist in range(1, 3):
|
|
+ # sugs will return suggest first ordered by levenshtein distance
|
|
+ # then second order by dictionary order
|
|
+ target += [w for w in _wordlist
|
|
+ if levenshtein(typo, w) <= lev_dist and w not in target]
|
|
assert sugs == target
|
|
|
|
|