def test_get_inter_mask(): lengths = np.array([5, 5]) mask = get_inter_mask(lengths, resolution=1) true_mask = np.zeros((10, 10)) true_mask[:5, :5] = 1 true_mask[5:, 5:] = 1 assert_array_equal(mask, true_mask.astype(bool) == False)
def test_get_inter_mask(): lengths = np.array([5, 5]) mask = get_inter_mask(lengths) true_mask = np.zeros((10, 10)) true_mask[:5, :5] = 1 true_mask[5:, 5:] = 1 assert_array_equal(mask, np.invert(true_mask.astype(bool)))