def test_has_similar(self):
data = {'foo': 1, 'bar': [1,2]}
doc = SolrDocument(data)
self.assertTrue(doc.has_similar(mlt_fields=['foo']))
self.assertTrue(doc.has_similar(mlt_fields=['bar']))
self.assertTrue(doc.has_similar(mlt_fields=['foo','bar']))
self.assertTrue(doc.has_similar(mlt_fields=['foo','baz']))
self.assertFalse(doc.has_similar(mlt_fields=['baz']))
def test_highlights(self):
data = {'highlights': {'foo': ['bar','baz']}}
doc = SolrDocument(data)
self.assertTrue(doc.has_highlights())
self.assertTrue(doc.has_highlights('foo'))
self.assertFalse(doc.has_highlights('blah'))
self.assertEqual(doc.get_highlights('foo'), ['bar','baz'])
self.assertEqual(doc.get_highlights('blah'), None)
def test_counts(self):
doc = SolrDocument({'[citations]': {'num_citations':3, 'num_references':2}})
self.assertEqual(doc.get_citation_count(), 3)
self.assertEqual(doc.get_references_count(), 2)
