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
|
from beanie.odm.operators.update.array import (
AddToSet,
Pop,
Pull,
PullAll,
Push,
)
from tests.odm.models import Sample
def test_add_to_set():
q = AddToSet({Sample.integer: 2})
assert q == {"$addToSet": {"integer": 2}}
def test_pop():
q = Pop({Sample.integer: 2})
assert q == {"$pop": {"integer": 2}}
def test_pull():
q = Pull({Sample.integer: 2})
assert q == {"$pull": {"integer": 2}}
def test_push():
q = Push({Sample.integer: 2})
assert q == {"$push": {"integer": 2}}
def test_pull_all():
q = PullAll({Sample.integer: 2})
assert q == {"$pullAll": {"integer": 2}}
|