def test346Inclusive(self):
   expected_mask = np.array(
       [[1, 0, 0, 0],
        [1, 0, 0, 0],
        [1, 1, 0, 0],
        [1, 1, 0, 0],
        [1, 1, 1, 0],
        [1, 1, 1, 0]])
   mask = _gen_mask(num_blocks=3, n_in=4, n_out=6, mask_type="inclusive")
   self.assertAllEqual(expected_mask, mask)
Example #2
0
 def test346Inclusive(self):
     expected_mask = np.array([[1, 0, 0, 0], [1, 0, 0, 0], [1, 1, 0, 0],
                               [1, 1, 0, 0], [1, 1, 1, 0], [1, 1, 1, 0]])
     mask = _gen_mask(num_blocks=3, n_in=4, n_out=6, mask_type="inclusive")
     self.assertAllEqual(expected_mask, mask)