o tf@sdZddlmZddlZddlZddlZddlmZmZddl Z ddl m Z e j ddd Z e j dd d Ze j dddZe j dddZe j dddZdS)z%Fixtures for use with jupyter events.) annotationsN)AnyCallable) EventLoggerreturn io.StringIOcCstS)zA stream for capture events.)ioStringIOr r e/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/jupyter_events/pytest_plugin.py jp_event_sinksr logging.HandlercCs t|S)zGA logging handler that captures any events emitted by the event handler)logging StreamHandler)r r r r jp_event_handlers rCallable[..., list[str] | None]csdfdd }|S)z3Reads list of events since last time it was called.rlist[str] | NonecsH}|rdd|dDnd}dd|S)NcSsg|]}t|qSr )jsonloads).0itemr r r #sz9jp_read_emitted_events.._read.. r)flushgetvaluestripsplittruncateseek)Z event_bufoutputrr r r _read s    z%jp_read_emitted_events.._readN)rrr )rr r!r r r jp_read_emitted_eventss r" list[Any]cCsgS)zA list of schema references. Each item should be one of the following: - string of serialized JSON/YAML content representing a schema - a pathlib.Path object pointing to a schema file on disk - a dictionary with the schema data. r r r r r jp_event_schemas,s r$rcCs*t}|D]}||q|j|d|S)z(A pre-configured event logger for tests.)handler)rZregister_event_schemaZregister_handler)rr$loggerZschemar r r jp_event_logger8s   r')rr)r rrr )rr r rrr)rr#)rr r$r#rr)__doc__ __future__rrrrtypingrrZpytestZjupyter_eventsrZfixturer rr"r$r'r r r r s$