Error in Week 3 Assignment 2, Exercise 3 unit test error message

The unit test for the insert_ones function appears to give the wrong error message for the 5th assert statement

assert arr1[0][segment_end_y] == 0, f"Array at {segment_end_y} must be 1"

should presumably be

assert arr1[0][segment_end_y] == 0, f"Array at {segment_end_y} must be 0"

Not a major issue, but could be confusing for someone trying to debug.

For what it’s worth, it also seems that it’s possible to pass the unit test with a function that gets the final/maximum index wrong, and leaves the final index at 0 rather than setting it to 1. (Though this will show up in the plot on the next cell)