Add unit tests for _patch_label_2d()

[yalaudah]

In test.py, unit tests are missing for _patch_label_2d(). These tests will help us catch errors where predictions are all zeros or some other values.