def create_context(add_serializers=False, add_datadiff=False, add_extensions=False, **kwargs): context = yaql.create_context(**kwargs) if add_serializers: serializers.register(context) if add_datadiff: datadiff.register(context) if add_extensions: extensions.setup_yaql_context(context) return context
def test_setup_yaql_context(self, get_m): context = mock.Mock() setup_yaql_context(context) for ext in get_m.return_value: ext.setup_yaql_context.assert_called_once_with(context)