I will recommd the Python based lightweight 2d simulator ir-sim: https://github.com/hanruihua/ir-sim. It is very simple to use by configuring the environment using yaml file.