def test_has_uppercase_elements(self):
target = Target({})
actual = target.preparation("Red HEEL")
self.assertDictEqual(
actual,
{
'used_query': 'Red HEEL',
'original_query': 'Red HEEL',
'tokens': [
{
'start': 0,
'end': 3,
'pos': 'NNP',
'stem': u'red',
'stop_word': False,
'skip_word': False,
'use': True,
'value': 'red'
},
{
'start': 4,
'end': 8,
'pos': 'NNP',
'stem': u'heel',
'stop_word': False,
'skip_word': False,
'use': True,
'value': 'heel'
}
]
}
)
def test_has_uppercase_elements(self):
target = Target({})
actual = target.preparation("Red HEEL")
self.assertDictEqual(
actual, {
'used_query':
'Red HEEL',
'original_query':
'Red HEEL',
'tokens': [{
'start': 0,
'end': 3,
'pos': 'NNP',
'stem': u'red',
'stop_word': False,
'skip_word': False,
'use': True,
'value': 'red'
}, {
'start': 4,
'end': 8,
'pos': 'NNP',
'stem': u'heel',
'stop_word': False,
'skip_word': False,
'use': True,
'value': 'heel'
}]
})
def test_empty_string(self):
target = Target({})
actual = target.preparation("")
self.assertDictEqual(actual, {
'used_query': '',
'tokens': [],
'original_query': ''
})
def test_has_stopwords(self):
target = Target({})
actual = target.preparation("Shoes with red and white")
self.assertDictEqual(
actual,
{
'original_query': 'Shoes with red and white',
'used_query': 'Shoes with red and white',
'tokens': [
{
'start': 0,
'end': 5,
'pos': 'VBZ',
'stem': u'shoe',
'stop_word': False,
'skip_word': False,
'use': True,
'value': 'shoes'},
{
'start': 6,
'end': 10,
'pos': 'IN',
'stem': u'with',
'stop_word': True,
'skip_word': False,
'use': False,
'value': 'with'},
{
'start': 11,
'end': 14,
'pos': 'JJ',
'stem': u'red',
'stop_word': False,
'skip_word': False,
'use': True,
'value': 'red'},
{
'start': 15,
'end': 18,
'pos': 'CC',
'stem': u'and',
'stop_word': True,
'skip_word': False,
'use': False,
'value': 'and'},
{
'start': 19,
'end': 24,
'pos': 'JJ',
'stem': u'white',
'stop_word': False,
'skip_word': False,
'use': True,
'value': 'white'
}
]
}
)
def test_empty_string(self):
target = Target({})
actual = target.preparation("")
self.assertDictEqual(
actual,
{
'used_query': '',
'tokens': [],
'original_query': ''
}
)
def test_has_skipwords(self):
target = Target({})
actual = target.preparation("Show me anything")
self.assertDictEqual(
actual,
{
'original_query': 'Show me anything',
'used_query': 'Show me anything',
'tokens': [
{
'start': 0,
'end': 4,
'pos': 'NNP',
'stem': u'show',
'stop_word': True,
'skip_word': True,
'use': False,
'value': 'show'},
{
'start': 5,
'end': 7,
'pos': 'PRP',
'stem': u'me',
'stop_word': True,
'skip_word': False,
'use': False,
'value': 'me'},
{
'start': 8,
'end': 16,
'pos': 'NN',
'stem': u'anyth',
'stop_word': False,
'skip_word': True,
'use': False,
'value': 'anything'
}
]
}
)
