TST: unit test for setting keys in the mask