R Under development (unstable) (2023-07-19 r84711 ucrt) -- "Unsuffered Consequences"
Copyright (C) 2023 The R Foundation for Statistical Computing
Platform: x86_64-w64-mingw32/x64

R is free software and comes with ABSOLUTELY NO WARRANTY.
You are welcome to redistribute it under certain conditions.
Type 'license()' or 'licence()' for distribution details.

R is a collaborative project with many contributors.
Type 'contributors()' for more information and
'citation()' on how to cite R or R packages in publications.

Type 'demo()' for some demos, 'help()' for on-line help, or
'help.start()' for an HTML browser interface to help.
Type 'q()' to quit R.

> library(reclin2)
Loading required package: data.table

Attaching package: 'reclin2'

The following object is masked from 'package:base':

    identical

> library(parallel)
> source("helpers.R")
> 
> pairs <- data.table(
+   .x  = c(3,3,1,2,3,2,4), 
+   .y  = c(4,5,1,2,1,3,6),
+   sel = c(1,1,1,1,1,1,1) == 1
+ )
> class(pairs) <- c("pairs", class(pairs))
> 
> pairs$sel <- c(1,1,1,1,1,1,1) == 1
> tmp <- select_unique(pairs, "sel2", preselect = "sel")
> expect_equal(names(tmp), c(names(pairs), "sel2"))
> expect_equal(tmp$sel2, c(0,0,0,0,0,0,1) == 1)
> 
> pairs$sel <- c(1,1,1,1,1,1,1) == 1
> tmp <- select_unique(pairs, "sel2")
> expect_equal(names(tmp), c(names(pairs), "sel2"))
> expect_equal(tmp$sel2, c(0,0,0,0,0,0,1) == 1)
> 
> pairs$sel <- c(1,0,1,1,0,1,1) == 1
> tmp <- select_unique(pairs, "sel2")
> expect_equal(tmp$sel2, c(0,0,0,0,0,0,1) == 1)
> tmp <- select_unique(pairs, "sel2", preselect = "sel")
> expect_equal(tmp$sel2, c(1,0,1,0,0,0,1) == 1)
> 
> pairs$sel <- c(0,0,0,0,0,0,0) == 1
> tmp <- select_unique(pairs, "sel2")
> expect_equal(tmp$sel2, c(0,0,0,0,0,0,1) == 1)
> tmp <- select_unique(pairs, "sel2", preselect = "sel")
> expect_equal(tmp$sel2, c(0,0,0,0,0,0,0) == 1)
> 
> 
> # === N & M ARGUMENTS
> pairs$sel <- c(1,1,1,1,1,1,1) == 1
> tmp <- select_unique(pairs, "sel2", preselect = "sel", n = Inf)
> expect_equal(tmp$sel2, c(0,0,1,0,0,0,1) == 1)
> tmp <- select_unique(pairs, "sel2", preselect = "sel", m = Inf)
> expect_equal(tmp$sel2, c(1,1,0,1,0,1,1) == 1)
> tmp <- select_unique(pairs, "sel2", preselect = "sel", m = 2)
> expect_equal(tmp$sel2, c(0,0,0,1,0,1,1) == 1)
> tmp <- select_unique(pairs, "sel2", preselect = "sel", n = 2, m = 2)
> expect_equal(tmp$sel2, c(0,0,1,1,0,1,1) == 1)
> tmp <- select_unique(pairs, "sel2", preselect = "sel", n = Inf, m = Inf)
> expect_equal(tmp$sel2, c(1,1,1,1,1,1,1) == 1)
> 
> # === INPLACE ARGUMENT
> pairs$sel <- c(1,1,1,1,1,1,1) == 1
> select_unique(pairs, "sel2", preselect = "sel", n = Inf, inplace = TRUE)
> expect_equal(pairs$sel2, c(0,0,1,0,0,0,1) == 1)
> pairs[, sel2 := NULL]
  Total number of pairs: 7 pairs

   .x .y  sel
1:  3  4 TRUE
2:  3  5 TRUE
3:  1  1 TRUE
4:  2  2 TRUE
5:  3  1 TRUE
6:  2  3 TRUE
7:  4  6 TRUE
> 
> # === ID_X and ID_Y
> x <- data.table(id = c(1,2,3,4))
> setattr(pairs, "x", x)
> y <- data.table(id = c(1,2,3,4,5,6))
> setattr(pairs, "y", y)
> 
> pairs$sel <- c(1,1,1,1,1,1,1) == 1
> tmp <- select_unique(pairs, "sel2", preselect = "sel", 
+   id_x = "id", id_y = "id")
> expect_equal(tmp$sel2, c(0,0,0,0,0,0,1) == 1)
> 
> x <- data.table(id = c(1,2,3,4))
> setattr(pairs, "x", x)
> y <- data.table(id = c(1,2,3,4,5,5))
> setattr(pairs, "y", y)
> 
> pairs$sel <- c(1,1,1,1,1,1,1) == 1
> tmp <- select_unique(pairs, "sel2", preselect = "sel", 
+   id_x = "id", id_y = "id")
> expect_equal(tmp$sel2, c(0,0,0,0,0,0,0) == 1)
> 
> id_x <- x$id[pairs$.x]
> id_y <- x$id[pairs$.y]
> tmp <- select_unique(pairs, "sel2", preselect = "sel", 
+   id_x = id_x, id_y = id_y)
> expect_equal(tmp$sel2, c(0,0,0,0,0,0,0) == 1)
> 
> expect_error(
+ tmp <- select_unique(pairs, "sel2", preselect = "sel", 
+   id_x = 1:2, id_y = id_y)
+ )
> expect_error(
+ tmp <- select_unique(pairs, "sel2", preselect = "sel", 
+   id_x = id_x, id_y = 1:3)
+ )
> 
> # === EDGE CASES
> # Empty pairs
> pairs$sel <- c(1,1,1,1,1,1,1) == 1
> tmp <- select_unique(pairs[FALSE, ], "sel2", preselect = "sel", n = Inf)
> expect_equal(names(tmp), c(names(pairs), "sel2"))
> expect_equal(tmp$sel2, logical(0))
> 
> # === CLUSTER
> x <- data.table(a = c(1,1,2,2), b = c(1,2,1,2))
> y <- data.table(a = c(3,3,2,2), b = c(1,2,1,2))
> 
> cl <- makeCluster(2)
> clusterEvalQ(cl, data.table::setDTthreads(1))
[[1]]
[1] 2

[[2]]
[1] 2

> 
> set.seed(103)
> pairs <- cluster_pair_blocking(cl, x, y, on = "a")
> compare_pairs(pairs, c("a", "b"))
> score_simple(pairs, on = c("a", "b"), variable = "score")
> 
> expect_error(
+   select_unique(pairs, "sel2")
+ )
> 
> stopCluster(cl)
> 
> proc.time()
   user  system elapsed 
   0.34    0.04    0.76