File: test_array.py

package info (click to toggle)
python-beanie 2.0.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 1,480 kB
  • sloc: python: 14,427; makefile: 7; sh: 6
file content (28 lines) | stat: -rw-r--r-- 730 bytes parent folder | download | duplicates (2)
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
from beanie.odm.operators.find.array import All, ElemMatch, Size
from tests.odm.models import PackageElemMatch, Sample


async def test_all():
    q = All(Sample.integer, [1, 2, 3])
    assert q == {"integer": {"$all": [1, 2, 3]}}


async def test_elem_match():
    q = ElemMatch(Sample.integer, {"a": "b"})
    assert q == {"integer": {"$elemMatch": {"a": "b"}}}


async def test_size():
    q = Size(Sample.integer, 4)
    assert q == {"integer": {"$size": 4}}


async def test_elem_match_nested():
    q = ElemMatch(
        PackageElemMatch.releases, major_ver=7, minor_ver=1, build_ver=0
    )
    assert q == {
        "releases": {
            "$elemMatch": {"major_ver": 7, "minor_ver": 1, "build_ver": 0}
        }
    }