import jax.numpy as jnp
import matplotlib.pyplot as plt

from differt2d.geometry import Wall
from differt2d.scene import Scene

ax = plt.gca()
scene = Scene.square_scene()
wall = Wall(xys=jnp.array([[.8, .2], [.8, .8]]))
scene = scene.add_objects(wall)
scene.plot(ax)
plt.show()