python_arango_ogm.db.tests.test_pao_models 

 1from python_arango_ogm.db.tests.database import use_database
 2from python_arango_ogm.db.tests.models import FooModel, BarModel, BazModel
 3
 4
 5def test_model_basic():
 6    fields = BarModel.get_fields()
 7    assert ("field_str" in fields)
 8    assert ("field_int" in fields)
 9
10
11def test_models_with_named_edge():
12    fields = FooModel.get_fields()
13    assert ("field_int" in fields)
14    assert ("field_str" in fields)
15    assert ("bar_edge" in fields)
16
17
18def test_models_with_edge_obj():
19    fields = BazModel.get_fields()
20    assert ("field_str" in fields)
21    assert ("field_int" in fields)
22    assert ("foo_edge" in fields)
23
24
25def test_model_marshalling():
26    with use_database() as db:
27        for i in range(10):
28            FooModel.insert({"field_str": f"foo_{i}", "field_int": i})
29            BarModel.insert({"field_str": f"bar_{i}", "field_int": i})
30
31        foos = FooModel.all(sort_fields={"field_str": "ASC"})
32        print("FOOS:", foos)
33        assert len(foos) == 10
34
35
36if __name__ == '__main__':
37    test_model_marshalling()
38    # test_migrator_idempotency()
def test_model_basic(): 
6def test_model_basic():
7    fields = BarModel.get_fields()
8    assert ("field_str" in fields)
9    assert ("field_int" in fields)
def test_models_with_named_edge(): 
12def test_models_with_named_edge():
13    fields = FooModel.get_fields()
14    assert ("field_int" in fields)
15    assert ("field_str" in fields)
16    assert ("bar_edge" in fields)
def test_models_with_edge_obj(): 
19def test_models_with_edge_obj():
20    fields = BazModel.get_fields()
21    assert ("field_str" in fields)
22    assert ("field_int" in fields)
23    assert ("foo_edge" in fields)
def test_model_marshalling(): 
26def test_model_marshalling():
27    with use_database() as db:
28        for i in range(10):
29            FooModel.insert({"field_str": f"foo_{i}", "field_int": i})
30            BarModel.insert({"field_str": f"bar_{i}", "field_int": i})
31
32        foos = FooModel.all(sort_fields={"field_str": "ASC"})
33        print("FOOS:", foos)
34        assert len(foos) == 10