specify k_best at function call

irshadbhat · Jul 2, 2016 · cdd336e · cdd336e
1 parent 55bcd96
commit cdd336e
Showing 1 changed file with 10 additions and 12 deletions.
diff --git a/indictrans/tests/test_trans.py b/indictrans/tests/test_trans.py
@@ -43,18 +43,16 @@ def test_trg2src(self):
 
     def test_kbest(self):
         k_best = range(2, 15)
+        r2i = transliterator(
+            source='eng',
+            target='hin',
+            decode='beamsearch')
+        i2r = transliterator(
+            source='hin',
+            target='eng',
+            decode='beamsearch')
         for k in k_best:
-            r2i = transliterator(
-                source='eng',
-                target='hin',
-                decode='beamsearch',
-                k_best=k)
-            i2r = transliterator(
-                source='hin',
-                target='eng',
-                decode='beamsearch',
-                k_best=k)
-            hin = r2i.transform('indictrans')
-            eng = i2r.transform(hin[0])
+            hin = r2i.transform('indictrans', k_best=k)
+            eng = i2r.transform(hin[0], k_best=k)
             assert len(hin) == k
             assert len(eng) == k