o ^i@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 W/home/ubuntu/hpml_nyu/venv/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) 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)rregister_event_schemaregister_handler)rr%loggerschemar r r jp_event_logger8s   r+)rr)r rrr )rr r rrr)rr$)rr r%r$rr)__doc__ __future__rrrrtypingrrpytestjupyter_eventsrfixturer rr#r%r+r r r r s$