run black to please ci

This commit is contained in:
Bastien Gérard 2020-08-11 21:55:22 +02:00
parent a5302b870b
commit 7116dec74a
2 changed files with 5 additions and 10 deletions

View File

@ -774,7 +774,7 @@ class EmbeddedDocumentField(BaseField):
def prepare_query_value(self, op, value):
if value is not None and not isinstance(value, self.document_type):
# Short circuit for special operators, returning them as is
if isinstance(value, dict) and all(k.startswith('$') for k in value.keys()):
if isinstance(value, dict) and all(k.startswith("$") for k in value.keys()):
return value
try:
value = self.document_type._from_son(value)

View File

@ -352,24 +352,19 @@ class TestTransform(unittest.TestCase):
class Drink(EmbeddedDocument):
id = StringField()
meta = {
'strict': False
}
meta = {"strict": False}
class Shop(Document):
drinks = EmbeddedDocumentListField(Drink)
Shop.drop_collection()
drinks = [Drink(id='drink_1'), Drink(id='drink_2')]
drinks = [Drink(id="drink_1"), Drink(id="drink_2")]
Shop.objects.create(drinks=drinks)
q_obj = transform.query(
Shop,
drinks__all=[{'$elemMatch': {'_id': x.id}} for x in drinks]
Shop, drinks__all=[{"$elemMatch": {"_id": x.id}} for x in drinks]
)
assert q_obj == {
'drinks': {
'$all': [{'$elemMatch': {'_id': x.id}} for x in drinks]
}
"drinks": {"$all": [{"$elemMatch": {"_id": x.id}} for x in drinks]}
}
Shop.drop_collection()