a Kb@s2ddlmZddlZddlmZGdddZdS))TupleN) edit_distancec@s:eZdZejdgdeeeeeefdddZ dS)TestEditDistancez&left,right,substitution_cost,expecteds))abcca)rrr)wantswasprr r )r r r r)rainshinerr)rrr r r )acbdefabcdefrrr )rrr r)lnaguaeglanguagerr )rrr r)lnaugagerrr)rrr r)lngauagerrr r )rrr r)r swimrr)r rr ))kittensittingrr)rr r r) duplicated duuplicatedrrr)r!r"r r#)zvery duplicatedzvery duuplicateedr r)leftrightsubstitution_cost expectedsc CsR||f||ffD]<\}}t|ddgD]$\}}t||||d} | |ks&Jq&qdS)a\ Test `edit_distance` between two strings, given some `substitution_cost`, and whether transpositions are allowed. :param str left: First input string to `edit_distance`. :param str right: Second input string to `edit_distance`. :param int substitution_cost: The cost of a substitution action in `edit_distance`. :param Tuple[int, int] expecteds: A tuple of expected outputs, such that `expecteds[0]` is the expected output with `transpositions=True`, and `expecteds[1]` is the expected output with `transpositions=False`. TF)r&transpositionsN)zipr) selfr$r%r&r's1s2Zexpectedr(Z predictedr-;lib/python3.9/site-packages/nltk/test/unit/test_distance.pytest_with_transpositions soz)TestEditDistance.test_with_transpositionsN) __name__ __module__ __qualname__pytestZmarkZ parametrizestrintrr/r-r-r-r.rs ar)typingrr3Znltk.metrics.distancerrr-r-r-r.s