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)