def test_get_numbers_from_tokens_works_with_magnitude_words(self): tokens = [Token(x) for x in ['one', 'million', '7', 'thousand']] numbers = TableQuestionKnowledgeGraph._get_numbers_from_tokens(tokens) assert numbers == [('1000000', 'one million'), ('7000', '7 thousand')]
def test_get_numbers_from_tokens_works_for_months(self): tokens = [Token(x) for x in ['January', 'March', 'october']] numbers = TableQuestionKnowledgeGraph._get_numbers_from_tokens(tokens) assert numbers == [('1', 'January'), ('3', 'March'), ('10', 'october')]
def test_get_numbers_from_tokens_works_for_units(self): tokens = [Token(x) for x in ['1ghz', '3.5mm', '-2m/s']] numbers = TableQuestionKnowledgeGraph._get_numbers_from_tokens(tokens) assert numbers == [('1', '1ghz'), ('3.500', '3.5mm'), ('-2', '-2m/s')]
def test_get_numbers_from_tokens_works_for_arabic_numerals(self): tokens = [Token(x) for x in ['7', '1.0', '-20']] numbers = TableQuestionKnowledgeGraph._get_numbers_from_tokens(tokens) assert numbers == [('7', '7'), ('1.000', '1.0'), ('-20', '-20')]
def test_get_numbers_from_tokens_works_for_ordinal_and_cardinal_numbers( self): tokens = [Token(x) for x in ['one', 'five', 'Seventh']] numbers = TableQuestionKnowledgeGraph._get_numbers_from_tokens(tokens) assert numbers == [('1', 'one'), ('5', 'five'), ('7', 'Seventh')]
def test_get_numbers_from_tokens_works_for_ordinal_and_cardinal_numbers(self): tokens = [Token(x) for x in ['one', 'five', 'Seventh']] numbers = TableQuestionKnowledgeGraph._get_numbers_from_tokens(tokens) assert numbers == [('1', 'one'), ('5', 'five'), ('7', 'Seventh')]