Add tests for single unit operation

A but was discovered in #28, where the solution was not returned correctly if the simulation only contained a single unit operation. This was fixed but unfortunately, we currently don't have tests for single-unit-operation systems (because currently we mainly use the LWE function).

We should improve this in the future.

@ronald-jaepel since you reported the issue, could you please provide a MRE?
