Add tests to assign_model_colours.py #1882
Description
What problem does your feature request solve?
We recently had a nasty bug in assign_model_colours.py (#1846) that could have been caught by unit tests. We currently have none for this file.
Describe the solution you'd like
We should add tests for this code.
Describe alternatives you've considered
Live life on the edge?