1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
|
from beanie.odm.operators.find.comparison import (
GT,
GTE,
LT,
LTE,
NE,
Eq,
In,
NotIn,
)
from tests.odm.models import Sample
async def test_eq():
q = Sample.integer == 1
assert q == {"integer": 1}
q = Eq(Sample.integer, 1)
assert q == {"integer": 1}
q = Eq("integer", 1)
assert q == {"integer": 1}
async def test_gt():
q = Sample.integer > 1
assert q == {"integer": {"$gt": 1}}
q = GT(Sample.integer, 1)
assert q == {"integer": {"$gt": 1}}
q = GT("integer", 1)
assert q == {"integer": {"$gt": 1}}
async def test_gte():
q = Sample.integer >= 1
assert q == {"integer": {"$gte": 1}}
q = GTE(Sample.integer, 1)
assert q == {"integer": {"$gte": 1}}
q = GTE("integer", 1)
assert q == {"integer": {"$gte": 1}}
async def test_in():
q = In(Sample.integer, [1])
assert q == {"integer": {"$in": [1]}}
q = In(Sample.integer, [1])
assert q == {"integer": {"$in": [1]}}
async def test_lt():
q = Sample.integer < 1
assert q == {"integer": {"$lt": 1}}
q = LT(Sample.integer, 1)
assert q == {"integer": {"$lt": 1}}
q = LT("integer", 1)
assert q == {"integer": {"$lt": 1}}
async def test_lte():
q = Sample.integer <= 1
assert q == {"integer": {"$lte": 1}}
q = LTE(Sample.integer, 1)
assert q == {"integer": {"$lte": 1}}
q = LTE("integer", 1)
assert q == {"integer": {"$lte": 1}}
async def test_ne():
q = Sample.integer != 1
assert q == {"integer": {"$ne": 1}}
q = NE(Sample.integer, 1)
assert q == {"integer": {"$ne": 1}}
q = NE("integer", 1)
assert q == {"integer": {"$ne": 1}}
async def test_nin():
q = NotIn(Sample.integer, [1])
assert q == {"integer": {"$nin": [1]}}
q = NotIn(Sample.integer, [1])
assert q == {"integer": {"$nin": [1]}}
|