I was just thinking that commenting and uncommenting which code to use could be improved a bit. I have not looked at what test_bench.py did so I'm not sure of the relation. I was thinking the solution could be a switch-case function, kinda like in other languages, but apparently python doesn't have something like that built in, so I just searched up a way to do it, and they basically used a dictionary to match what you wanted with what to call. Just a small change :)
I was just thinking that commenting and uncommenting which code to use could be improved a bit. I have not looked at what test_bench.py did so I'm not sure of the relation. I was thinking the solution could be a switch-case function, kinda like in other languages, but apparently python doesn't have something like that built in, so I just searched up a way to do it, and they basically used a dictionary to match what you wanted with what to call. Just a small change :)