def test_relabel_sequential_nonpositive_offset(data_already_sequential,
offset):
if data_already_sequential:
ar = cp.asarray([1, 3, 0, 2, 5, 4])
else:
ar = cp.asarray([1, 1, 5, 5, 8, 99, 42, 0])
with pytest.raises(ValueError):
relabel_sequential(ar, offset=offset)
def test_arraymap_len():
ar = cp.asarray([1, 1, 5, 5, 8, 99, 42, 0], dtype=cp.intp)
relabeled, fw, inv = relabel_sequential(ar)
assert len(fw) == 100
assert len(fw) == len(cp.asarray(fw))
assert len(inv) == 6
assert len(inv) == len(cp.asarray(inv))
def test_relabel_sequential_int_dtype_stability(data_already_sequential,
dtype):
if data_already_sequential:
ar = cp.asarray([1, 3, 0, 2, 5, 4], dtype=dtype)
else:
ar = cp.asarray([1, 1, 5, 5, 8, 99, 42, 0], dtype=dtype)
assert all(a.dtype == dtype for a in relabel_sequential(ar))
def test_relabel_sequential_signed_overflow():
imax = cp.iinfo(cp.int32).max
labels = cp.asarray([0, 1, 99, 42, 42], dtype=cp.int32)
output, fw, inv = relabel_sequential(labels, offset=imax)
reference = np.array([0, imax, imax + 2, imax + 1, imax + 1],
dtype=cp.uint32)
assert_array_equal(output, reference)
assert output.dtype == reference.dtype
def test_relabel_sequential_int_dtype_overflow():
ar = cp.asarray([1, 3, 0, 2, 5, 4], dtype=cp.uint8)
offset = 254
ar_relab, fw, inv = relabel_sequential(ar, offset=offset)
_check_maps(ar, ar_relab, fw, inv)
assert all(a.dtype == cp.uint16 for a in (ar_relab, fw))
assert inv.dtype == ar.dtype
ar_relab_ref = cp.where(ar > 0, ar.astype(cp.int) + offset - 1, 0)
assert_array_equal(ar_relab, ar_relab_ref)
def test_relabel_sequential_dtype():
ar = cp.asarray([1, 1, 5, 5, 8, 99, 42, 0], dtype=cp.uint8)
ar_relab, fw, inv = relabel_sequential(ar, offset=5)
_check_maps(ar.astype(int), ar_relab, fw, inv)
ar_relab_ref = np.array([5, 5, 6, 6, 7, 9, 8, 0])
assert_array_equal(ar_relab, ar_relab_ref)
fw_ref = np.zeros(100, int)
fw_ref[1] = 5
fw_ref[5] = 6
fw_ref[8] = 7
fw_ref[42] = 8
fw_ref[99] = 9
assert_array_equal(fw, fw_ref)
inv_ref = np.array([0, 0, 0, 0, 0, 1, 5, 8, 42, 99])
assert_array_equal(inv, inv_ref)
def test_relabel_sequential_offset1():
ar = cp.asarray([1, 1, 5, 5, 8, 99, 42])
ar_relab, fw, inv = relabel_sequential(ar)
_check_maps(ar, ar_relab, fw, inv)
ar_relab_ref = np.array([1, 1, 2, 2, 3, 5, 4])
assert_array_equal(ar_relab, ar_relab_ref)
fw_ref = np.zeros(100, int)
fw_ref[1] = 1
fw_ref[5] = 2
fw_ref[8] = 3
fw_ref[42] = 4
fw_ref[99] = 5
assert_array_equal(fw, fw_ref)
inv_ref = np.array([0, 1, 5, 8, 42, 99])
assert_array_equal(inv, inv_ref)
def test_relabel_sequential_already_sequential(offset, with0,
input_starts_at_offset):
if with0:
ar = cp.asarray([1, 3, 0, 2, 5, 4])
else:
ar = cp.asarray([1, 3, 2, 5, 4])
if input_starts_at_offset:
ar[ar > 0] += offset - 1
ar_relab, fw, inv = relabel_sequential(ar, offset=offset)
_check_maps(ar, ar_relab, fw, inv)
if input_starts_at_offset:
ar_relab_ref = ar
else:
ar_relab_ref = cp.where(ar > 0, ar + offset - 1, 0)
assert_array_equal(ar_relab, ar_relab_ref)
def test_arraymap_set():
ar = cp.asarray([1, 1, 5, 5, 8, 99, 42, 0], dtype=cp.intp)
relabeled, fw, inv = relabel_sequential(ar)
fw[72] = 6
assert fw[72] == 6
def test_arraymap_call():
ar = cp.asarray([1, 1, 5, 5, 8, 99, 42, 0], dtype=cp.intp)
relabeled, fw, inv = relabel_sequential(ar)
assert_array_equal(relabeled, fw(ar))
assert_array_equal(ar, inv(relabeled))
def test_incorrect_input_dtype():
labels = cp.asarray([0, 2, 2, 1, 1, 8], dtype=float)
with pytest.raises(TypeError):
relabel_sequential(labels)
def test_relabel_sequential_negative_values():
ar = cp.asarray([1, 1, 5, -5, 8, 99, 42, 0])
with pytest.raises(ValueError):
relabel_sequential(ar)
def test_very_large_labels():
imax = cp.iinfo(cp.int64).max
labels = cp.asarray([0, 1, imax, 42, 42], dtype=cp.int64)
output, fw, inv = relabel_sequential(labels, offset=imax)
assert int(cp.max(output)) == imax + 2