def test_gte():
from flamingo.core.data_model import ContentSet
cs = ContentSet()
for i in range(10):
cs.add(a=i)
cs = cs.filter(a__gte=5)
assert [i['a'] for i in cs] == [5, 6, 7, 8, 9]
def test_in():
from flamingo.core.data_model import ContentSet
cs = ContentSet()
for i in range(10):
cs.add(a=i)
cs = cs.filter(a__in=[4, 5, 8])
assert [i['a'] for i in cs] == [4, 5, 8]
def test_lt():
from flamingo.core.data_model import ContentSet
cs = ContentSet()
for i in range(10):
cs.add(a=i)
cs = cs.filter(a__lt=5)
assert [i['a'] for i in cs] == [0, 1, 2, 3, 4]
def test_filter():
from flamingo.core.data_model import ContentSet
# keyword args
cs = ContentSet()
for i in range(10):
cs.add(a=i)
cs = cs.filter(a=5)
assert cs.count() == 1
assert cs[0]['a'] == 5
# dict
cs = ContentSet()
for i in range(10):
cs.add(a=i)
cs = cs.filter({'a': 5})
assert cs.count() == 1
assert cs[0]['a'] == 5
