def test_duplicated_with_nas():
keys = [0, 1, np.nan, 0, 2, np.nan]
result = lib.duplicated(keys)
expected = [False, False, False, True, False, True]
assert (np.array_equal(result, expected))
result = lib.duplicated(keys, take_last=True)
expected = [True, False, True, False, False, False]
assert (np.array_equal(result, expected))
keys = [(0, 0), (0, np.nan), (np.nan, 0), (np.nan, np.nan)] * 2
result = lib.duplicated(keys)
falses = [False] * 4
trues = [True] * 4
expected = falses + trues
assert (np.array_equal(result, expected))
result = lib.duplicated(keys, take_last=True)
expected = trues + falses
assert (np.array_equal(result, expected))
def test_duplicated_with_nas():
keys = [0, 1, np.nan, 0, 2, np.nan]
result = lib.duplicated(keys)
expected = [False, False, False, True, False, True]
assert(np.array_equal(result, expected))
result = lib.duplicated(keys, take_last=True)
expected = [True, False, True, False, False, False]
assert(np.array_equal(result, expected))
keys = [(0, 0), (0, np.nan), (np.nan, 0), (np.nan, np.nan)] * 2
result = lib.duplicated(keys)
falses = [False] * 4
trues = [True] * 4
expected = falses + trues
assert(np.array_equal(result, expected))
result = lib.duplicated(keys, take_last=True)
expected = trues + falses
assert(np.array_equal(result, expected))
def test_duplicated_with_nas():
keys = np.array([0, 1, nan, 0, 2, nan], dtype=object)
result = lib.duplicated(keys)
expected = [False, False, False, True, False, True]
assert(np.array_equal(result, expected))
result = lib.duplicated(keys, take_last=True)
expected = [True, False, True, False, False, False]
assert(np.array_equal(result, expected))
keys = np.empty(8, dtype=object)
for i, t in enumerate(zip([0, 0, nan, nan]*2, [0, nan, 0, nan]*2)):
keys[i] = t
result = lib.duplicated(keys)
falses = [False] * 4
trues = [True] * 4
expected = falses + trues
assert(np.array_equal(result, expected))
result = lib.duplicated(keys, take_last=True)
expected = trues + falses
assert(np.array_equal(result, expected))