"""Tests for Yelp snippet creation.

These tests are meant to be run using nosetests.

"""

import snippets

class TestFullMatch(object):

assert snippet == '[[HIGHLIGHT]]pepperoni pizza[[ENDHIGHLIGHT]]'

class TestSingleSentence(object):

assert snippet == '[[HIGHLIGHT]]pizza[[ENDHIGHLIGHT]] [[HIGHLIGHT]]pepperoni olive pizza[[ENDHIGHLIGHT]] [[HIGHLIGHT]]olive pizza[[ENDHIGHLIGHT]]'

class TestCaseSensitivity(object):

assert snippet == '[[HIGHLIGHT]]pEpPeRoNi PiZzA[[ENDHIGHLIGHT]]'

class TestMaxChars(object):

assert snippet == '[[HIGHLIGHT]]Pizza[[ENDHIGHLIGHT]]!'

class TestMaxSents(object):

assert snippet.count('.') == 2

class TestMaxCharsAndSents(object):

assert snippet in 'Dog!', 'Cat!'

#

# Testing "private" functions

#

class TestFindQuerySpans(object):

assert spans == [(0, 1), (4, 6), (7, 10)]

#

# Testing "private" string-utility functions

#

class TestSplitIntoSentences(object):

assert sentences == ['I want some Thai food']

class TestSplitIntoWords(object):

assert words == ['It', 'costs', '.47', '.']

class TestJoinWords(object):

assert joined == 'Whatever... Hey there.'