def test_find_abbreviation(self):
# Basic case
doc = self.nlp("abbreviation (abbrn)")
long = doc[0:1]
short = doc[2:3]
_, long_form = find_abbreviation(long, short)
assert long_form.text == "abbreviation"
# Hypenation and numbers within abbreviation
doc = self.nlp("abbreviation (ab-b9rn)")
long = doc[0:1]
short = doc[2:3]
_, long_form = find_abbreviation(long, short)
assert long_form.text == "abbreviation"
# No match
doc = self.nlp("abbreviation (aeb-b9rn)")
long = doc[0:1]
short = doc[2:3]
_, long_form = find_abbreviation(long, short)
assert long_form is None
# First letter must match start of word.
doc = self.nlp("aaaabbreviation (ab-b9rn)")
long = doc[0:1]
short = doc[2:3]
_, long_form = find_abbreviation(long, short)
assert long_form.text == "aaaabbreviation"
# Matching is greedy for first letter (are is not included).
doc = self.nlp("more words are considered aaaabbreviation (ab-b9rn)")
long = doc[0:5]
short = doc[6:7]
_, long_form = find_abbreviation(long, short)
assert long_form.text == "aaaabbreviation"
def test_containsQuotes(self):
# Straight double quote
QUOTES = ['"', "'", "‘", "’", "“", "”"]
doc = self.nlp('abbreviation ("abbrn")')
long = doc[0:1]
short = doc[2:5]
short_form = short.text
_, long_form = find_abbreviation(long, short)
assert contains(short_form, QUOTES) is True
# Straight single quote
doc = self.nlp("abbreviation ('abbrn')")
long = doc[0:1]
short = doc[2:5]
short_form = short.text
_, long_form = find_abbreviation(long, short)
assert contains(short_form, QUOTES) is True
# Opening and closing single quotes
doc = self.nlp("abbreviation (‘abbrn’)")
long = doc[0:1]
short = doc[2:5]
short_form = short.text
_, long_form = find_abbreviation(long, short)
assert contains(short_form, QUOTES) is True
# Opening and closing double quotes
doc = self.nlp("abbreviation (“abbrn”)")
long = doc[0:1]
short = doc[2:5]
short_form = short.text
_, long_form = find_abbreviation(long, short)
assert contains(short_form, QUOTES) is True
# No quotes
doc = self.nlp("abbreviation (abbrn)")
long = doc[0:1]
short = doc[2:5]
short_form = short.text
_, long_form = find_abbreviation(long, short)
assert contains(short_form, QUOTES) is False