def test_allowed_does_not_raise_TypeError(self): from jicbioimage.core.util.array import check_dtype ar = np.zeros((5,5), dtype=np.uint64) check_dtype(ar, np.uint64)
def test_check_dtype_with_allowed_as_list(self): from jicbioimage.core.util.array import check_dtype ar = np.zeros((5,5), dtype=np.uint64) check_dtype(ar, [np.uint8, np.uint16, np.uint64])
def test_disallowed_raises_TypeError(self): from jicbioimage.core.util.array import check_dtype ar = np.zeros((5,5), dtype=np.uint64) with self.assertRaises(TypeError): check_dtype(ar, np.uint8)