import matplotlib.pyplot as plt
import jax
from differt2d.scene import Scene

ax = plt.gca()
key = jax.random.PRNGKey(1234)
scene = Scene.random_uniform_scene(n_walls=5, key=key)
_ = scene.plot(ax)
plt.show()  # doctest: +SKIP