diff --git a/qcond/transformers/merge.py b/qcond/transformers/merge.py index 2bdbde6..511d388 100644 --- a/qcond/transformers/merge.py +++ b/qcond/transformers/merge.py @@ -190,7 +190,7 @@ def full_value(self): cur = cur.parent - return total_score / len(words), ' '.join(words) + return float(total_score) / len(words), ' '.join(words) def __repr__(self): return '<%s value:"%s", weight: %s%s>' % ( diff --git a/tests/test_query_condenser.py b/tests/test_query_condenser.py index ef1ebc2..46f7e3c 100644 --- a/tests/test_query_condenser.py +++ b/tests/test_query_condenser.py @@ -62,6 +62,6 @@ def test_distinct(self): "La Leggenda Di Korra" ]), [ 'the korra', - 'legend of korra', - 'la leggenda di korra' + 'la leggenda di korra', + 'legend of korra' ]) diff --git a/tests/test_transformers.py b/tests/test_transformers.py index 384bf89..0c84073 100644 --- a/tests/test_transformers.py +++ b/tests/test_transformers.py @@ -31,6 +31,20 @@ def test_run(self): 'apartment 23' ]) + self.assertSequenceEqual(self.merge.run([ + "The Legend of Korra", + "The Last Airbender The Legend of Korra", + "Avatar: The Legend of Korra", + "Legend of Korra", + "La Leggenda Di Korra" + ]), [ + 'the', + 'the korra', + 'avatar the legend of korra', + 'la leggenda di korra', + 'legend of korra' + ]) + def test_merge(self): pass @@ -55,3 +69,14 @@ def test_run(self): "Don't Trust the B in Apt 23", 'Dont Trust the Bitch in Apartment 23' ]) + + self.assertSequenceEqual(self.slice.run([ + "The Legend of Korra", + "The Last Airbender The Legend of Korra", + "Avatar: The Legend of Korra", + "Legend of Korra", + "La Leggenda Di Korra" + ]), [ + 'Legend of Korra', + 'La Leggenda Di Korra' + ])