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))