def test_query(self):
q = SolrSearcher().search().dismax()
raw_query = str(q)
self.assertTrue('q=%s' % quote_plus('*:*') in raw_query)
self.assertFalse('defType=dismax' in raw_query)
q = SolrSearcher().search('test query').dismax()
raw_query = str(q)
self.assertTrue('q=%s' % quote_plus('test query') in raw_query)
self.assertTrue('defType=dismax' in raw_query)
q = (
SolrSearcher().search(X(name='test') | X(name__startswith='test'))
.dismax()
.qf([('name', 10), ('keywords', 2)])
.bf((func.linear('rank',1,0) ^ 100) + func.recip(func.ms('NOW/HOUR', 'dt_created'), 3.16e-11, 1, 1))
.field_weight('name', 5)
)
raw_query = str(q)
self.assertTrue('q=%s' % quote_plus('(name:test OR name:test*)') in raw_query)
self.assertTrue('qf=%s' % quote_plus('name^5 keywords^2') in raw_query)
self.assertTrue('bf=%s' % quote_plus('linear(rank,1,0)^100 recip(ms(NOW/HOUR,dt_created),3.16e-11,1,1)') in raw_query)
def test_func(self):
f = func.recip(func.sum('rank', 10), 1, 1, 0)
self.assertEqual(str(f), 'recip(sum(rank,10),1,1,0)')
f = func.linear('rank',
func.sqrt(func.sum(func.pow('x'), func.pow('y'))), 10)
self.assertEqual(str(f), 'linear(rank,sqrt(sum(pow(x),pow(y))),10)')
f = func.if_(func.exists('name'), 100, 0)
self.assertEqual(str(f), 'if(exists(name),100,0)')
f = func.termfreq('name', "test term '111'")
self.assertEqual(str(f), "termfreq(name,'test term \\'111\\'')")
f = func.undefined('name', 'null')
self.assertEqual(str(f), 'undefined(name,null)')
f = func.exists('name', 5,
0) * 2 + func.ord('popularity') * 0.5 + func.recip(
func.rord('price'), 1, 1000, 1000) * 0.3
self.assertEqual(
str(f),
'exists(name,5,0)^2 ord(popularity)^0.5 recip(rord(price),1,1000,1000)^0.3'
)
f = func.min(func.geodist('store', 37.7, -122.4),
func.geodist('store', 39.7, -105))
self.assertEqual(
str(f), 'min(geodist(store,37.7,-122.4),geodist(store,39.7,-105))')
self.assertRaises(TypeError, lambda: func.min('a', 'b') * '10')
self.assertRaises(TypeError, lambda: func.min('a', 'b') + '10')
self.assertRaises(
TypeError, lambda: func.min('a', 'b') + func.max('a', 'b') + '10')
def test_func(self):
f = func.recip(func.sum('rank',10), 1, 1, 0)
self.assertEqual(str(f), 'recip(sum(rank,10),1,1,0)')
f = func.linear('rank', func.sqrt(func.sum(func.pow('x'), func.pow('y'))), 10)
self.assertEqual(str(f), 'linear(rank,sqrt(sum(pow(x),pow(y))),10)')
f = func.if_(func.exists('name'), 100, 0)
self.assertEqual(str(f), 'if(exists(name),100,0)')
f = func.termfreq('name', "test term '111'")
self.assertEqual(str(f), "termfreq(name,'test term \\'111\\'')")
f = func.undefined('name', 'null')
self.assertEqual(str(f), 'undefined(name,null)')
f = func.exists('name', 5, 0) * 2 + func.ord('popularity') * 0.5 + func.recip(func.rord('price'), 1, 1000, 1000) * 0.3
self.assertEqual(str(f), 'exists(name,5,0)^2 ord(popularity)^0.5 recip(rord(price),1,1000,1000)^0.3')
f = func.min(func.geodist('store', 37.7, -122.4), func.geodist('store', 39.7, -105))
self.assertEqual(str(f), 'min(geodist(store,37.7,-122.4),geodist(store,39.7,-105))')
self.assertRaises(TypeError, lambda: func.min('a', 'b') * '10')
self.assertRaises(TypeError, lambda: func.min('a', 'b') + '10')
self.assertRaises(TypeError, lambda: func.min('a', 'b') + func.max('a', 'b') + '10')
def test_query(self):
q = SolrSearcher().search().dismax()
raw_query = str(q)
self.assertTrue('q=%s' % quote_plus('*:*') in raw_query)
self.assertFalse('defType=dismax' in raw_query)
q = SolrSearcher().search('test query').dismax()
raw_query = str(q)
self.assertTrue('q=%s' % quote_plus('test query') in raw_query)
self.assertTrue('defType=dismax' in raw_query)
q = (SolrSearcher().search(
X(name='test') | X(name__startswith='test')).dismax().qf([
('name', 10), ('keywords', 2)
]).bf((func.linear('rank', 1, 0) ^ 100) +
func.recip(func.ms('NOW/HOUR', 'dt_created'), 3.16e-11, 1, 1)
).field_weight('name', 5))
raw_query = str(q)
self.assertTrue('q=%s' %
quote_plus('(name:test OR name:test*)') in raw_query)
self.assertTrue('qf=%s' % quote_plus('name^5 keywords^2') in raw_query)
self.assertTrue('bf=%s' % quote_plus(
'linear(rank,1,0)^100 recip(ms(NOW/HOUR,dt_created),3.16e-11,1,1)')
in raw_query)
def test_func(self):
f = func.linear('rank', func.sqrt(func.sum(func.pow('x'), func.pow('y'))), 10)
self.assertEqual(str(f), 'linear(rank,sqrt(sum(pow(x),pow(y))),10)')
f = func.if_(func.exists('name'), 100, 0)
self.assertEqual(str(f), 'if(exists(name),100,0)')
f = (func.exists('name', 5, 0) ^ 2) + (func.ord('popularity') ^ 0.5) + (func.recip(func.rord('price'), 1, 1000, 1000) ^ 0.3)
self.assertEqual(str(f), 'exists(name,5,0)^2 ord(popularity)^0.5 recip(rord(price),1,1000,1000)^0.3')
def test_func(self):
f = func.linear('rank',
func.sqrt(func.sum(func.pow('x'), func.pow('y'))), 10)
self.assertEqual(str(f), 'linear(rank,sqrt(sum(pow(x),pow(y))),10)')
f = func.if_(func.exists('name'), 100, 0)
self.assertEqual(str(f), 'if(exists(name),100,0)')
f = (func.exists('name', 5, 0)
^ 2) + (func.ord('popularity') ^ 0.5) + (
func.recip(func.rord('price'), 1, 1000, 1000) ^ 0.3)
self.assertEqual(
str(f),
'exists(name,5,0)^2 ord(popularity)^0.5 recip(rord(price),1,1000,1000)^0.3'
)
def test_local_params(self):
self.assertEqual(force_unicode(LocalParams({'cache': False})),
'{!cache=false}')
self.assertEqual(force_unicode(LocalParams(LocalParams({'cache': False}))),
'{!cache=false}')
self.assertEqual(force_unicode(LocalParams({'ex': 'tag'}, key='tag')),
'{!ex=tag key=tag}')
self.assertEqual(force_unicode(LocalParams({'ex': 'tag', 'key': 'tag'}, key='category')),
'{!ex=tag key=category}')
self.assertEqual(force_unicode(LocalParams(ex=['f1', 'f2'])),
'{!ex=f1,f2}')
self.assertEqual(force_unicode(LocalParams('frange', l=0, u=5)),
'{!frange l=0 u=5}')
self.assertEqual(force_unicode(LocalParams(['geofilt', ('d', 10), ('key', 'd10')])),
'{!geofilt d=10 key=d10}')
self.assertEqual(force_unicode(LocalParams({'type': 'join', 'from': 'id', 'to': 'manu_id'})),
'{!join from=id to=manu_id}')
self.assertEqual(force_unicode(LocalParams()), '')
self.assertEqual(force_unicode(LocalParams(None)), '')
self.assertEqual(force_unicode(LocalParams('dismax', v='OR test')),
"""{!dismax v='or test'}""")
self.assertEqual(force_unicode(LocalParams('dismax', v='"test"')),
"""{!dismax v='\\"test\\"'}""")
self.assertEqual(force_unicode(LocalParams('dismax', v='test\'')),
"""{!dismax v='test\\\\\''}""")
self.assertRaises(ValueError, LocalParams, '{dismax}', v='test')
self.assertRaises(ValueError, LocalParams, ['dismax', ('!v', 'test')])
self.assertEqual(
force_unicode(LocalParams('dismax', qf='name',
v=X(SafeUnicode('"nokia lumia"')) | X(SafeUnicode('"nokia n900"')))),
"""{!dismax qf=name v='(\\"nokia lumia\\" OR \\"nokia n900\\")'}""")
lp = LocalParams('dismax', bf=func.linear('rank', 100, 0), v='$q1')
lp.update(LocalParams(qf='name^10 description'))
lp.add('pf', 'name')
lp.add('ps', 2)
self.assertIn('type', lp)
self.assertIn('v', lp)
self.assertFalse('q' in lp)
self.assertEqual(lp['type'], 'dismax')
self.assertEqual(
force_unicode(lp),
"{!dismax bf='linear(rank,100,0)' v=$q1 qf='name^10 description' pf=name ps=2}")
def test_local_params(self):
self.assertEqual(force_unicode(LocalParams({'cache': False})),
'{!cache=false}')
self.assertEqual(
force_unicode(LocalParams(LocalParams({'cache': False}))),
'{!cache=false}')
self.assertEqual(force_unicode(LocalParams({'ex': 'tag'}, key='tag')),
'{!ex=tag key=tag}')
self.assertEqual(
force_unicode(
LocalParams({
'ex': 'tag',
'key': 'tag'
}, key='category')), '{!ex=tag key=category}')
self.assertEqual(force_unicode(LocalParams(ex=['f1', 'f2'])),
'{!ex=f1,f2}')
self.assertEqual(force_unicode(LocalParams('frange', l=0, u=5)),
'{!frange l=0 u=5}')
self.assertEqual(
force_unicode(LocalParams(['geofilt', ('d', 10), ('key', 'd10')])),
'{!geofilt d=10 key=d10}')
self.assertEqual(
force_unicode(
LocalParams({
'type': 'join',
'from': 'id',
'to': 'manu_id'
})), '{!join from=id to=manu_id}')
self.assertEqual(force_unicode(LocalParams()), '')
self.assertEqual(force_unicode(LocalParams(None)), '')
self.assertEqual(force_unicode(LocalParams('dismax', v='OR test')),
"""{!dismax v='or test'}""")
self.assertEqual(force_unicode(LocalParams('dismax', v='"test"')),
"""{!dismax v='\\"test\\"'}""")
self.assertEqual(force_unicode(LocalParams('dismax', v='test\'')),
"""{!dismax v='test\\\\\''}""")
self.assertRaises(ValueError, LocalParams, '{dismax}', v='test')
self.assertRaises(ValueError, LocalParams, ['dismax', ('!v', 'test')])
self.assertEqual(
force_unicode(
LocalParams('dismax',
qf='name',
v=X(SafeUnicode('"nokia lumia"'))
| X(SafeUnicode('"nokia n900"')))),
"""{!dismax qf=name v='(\\"nokia lumia\\" OR \\"nokia n900\\")'}"""
)
lp = LocalParams('dismax', bf=func.linear('rank', 100, 0), v='$q1')
lp.update(LocalParams(qf='name^10 description'))
lp.add('pf', 'name')
lp.add('ps', 2)
self.assertIn('type', lp)
self.assertIn('v', lp)
self.assertFalse('q' in lp)
self.assertEqual(lp['type'], 'dismax')
self.assertEqual(
force_unicode(lp),
"{!dismax bf='linear(rank,100,0)' v=$q1 qf='name^10 description' pf=name ps=2}"
)
