AttributeError: data when repeatedly calling same function

[The-Compiler]

With a trivial answer.ipynb like this:

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2382e31c-1117-46df-b981-81100e7f68f3",
   "metadata": {},
   "outputs": [],
   "source": [
    "def answer():\n",
    "    return 42"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}

which simply contains a single cell with this code:

def answer():
    return 42

calling that function via testbook repeatedly (500+ times) will fail. Doing this:

from testbook import testbook

@testbook('answer.ipynb', execute=True)
def test_func(tb):
    answer = tb.get("answer")
    for _ in range(500):
        assert answer() == 42

if __name__ == "__main__":
    test_func()    # runnable with pure Python instead of pytest

results in:

Traceback (most recent call last):
  File "/home/florian/tmp/testbook/.venv/lib/python3.10/site-packages/nbformat/_struct.py", line 128, in __getattr__
    result = self[key]
KeyError: 'data'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/home/florian/tmp/testbook/test_answer.py", line 10, in <module>
    test_func()
  File "/home/florian/tmp/testbook/testbook/testbook.py", line 63, in wrapper
    return func(self.client, *args, **kwargs)
  File "/home/florian/tmp/testbook/test_answer.py", line 7, in test_func
    assert answer() == 42
  File "/home/florian/tmp/testbook/testbook/reference.py", line 85, in __call__
    return self.tb.value(code)
  File "/home/florian/tmp/testbook/testbook/client.py", line 302, in value
    return outputs[0].data['application/json']['value']
  File "/home/florian/tmp/testbook/.venv/lib/python3.10/site-packages/nbformat/_struct.py", line 130, in __getattr__
    raise AttributeError(key)
AttributeError: data

here:

testbook/testbook/client.py

Line 302 in e2ae899

return outputs[0].data['application/json']['value']

Adding a print(outputs) there reveals why:

[
    {
        'output_type': 'stream',
        'name': 'stderr',
        'text': "/home/florian/tmp/testbook/.venv/lib/python3.10/site-packages/IPython/core/displayhook.py:275: 
UserWarning: Output cache limit (currently 1000 entries) hit.\nFlushing oldest 200 entries.\n  warn('Output cache 
limit (currently {sz} entries) hit.\\n'\n"
    },
    {
        'output_type': 'execute_result',
        'metadata': {'application/json': {'expanded': False, 'root': 'root'}},
        'data': {'text/plain': '<IPython.core.display.JSON object>', 'application/json': {'value': 42}},
        'execution_count': 1003
    }
]

coming from IPython's displayhook.

Maybe instead of using outputs[0] there, testbook should find the first execute_result output or so? Though I'm also wondering if there's a way for testbook to prevent that warning altogether.