diff --git a/graphene_django/tests/test_command.py b/graphene_django/tests/test_command.py index a281abbd2..f7325d569 100644 --- a/graphene_django/tests/test_command.py +++ b/graphene_django/tests/test_command.py @@ -46,7 +46,7 @@ class Query(ObjectType): open_mock.assert_called_once() handle = open_mock() - assert handle.write.called_once() + handle.write.assert_called_once() schema_output = handle.write.call_args[0][0] assert schema_output == dedent(