tests/testthat/testRedshiftClient.R
In keboola/backend-r-client: Keboola backend database driver
library('keboola.backend.r.client')
test_that("connectRedshift", {
driver <- BackendDriver$new()
expect_equal(
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA),
TRUE
)
expect_that(
driver$connectRedshift("invalid", RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA),
throws_error()
)
})
test_that("prepare", {
driver <- BackendDriver$new()
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ?", "baz"),
"SELECT * FROM foo WHERE bar = 'baz'"
)
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ?", "ba'z"),
"SELECT * FROM foo WHERE bar = 'ba''z'"
)
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ?", 42),
"SELECT * FROM foo WHERE bar = '42'"
)
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ? AND baz = ?", list(bar=42,baz=21)),
"SELECT * FROM foo WHERE bar = '42' AND baz = '21'"
)
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ? AND baz = ?", 42, 21),
"SELECT * FROM foo WHERE bar = '42' AND baz = '21'"
)
})
test_that("update", {
driver <- BackendDriver$new()
expect_that(
driver$update("CREATE TABLE foo (bar INTEGER);"),
throws_error()
)
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS foo CASCADE;")
expect_equal(
driver$update("CREATE TABLE foo (bar INTEGER);"),
TRUE
)
expect_that(
driver$update("CREATE TABLE foobar error;"),
throws_error()
)
})
test_that("update", {
driver <- BackendDriver$new()
expect_that(
driver$update("CREATE TABLE foo (bar INTEGER);"),
throws_error()
)
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS foo CASCADE;")
expect_equal(
driver$update("CREATE TABLE foo (bar INTEGER);"),
TRUE
)
expect_that(
driver$update("CREATE TABLE foobar error;"),
throws_error()
)
})
test_that("tableExists", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS foo CASCADE;")
driver$update("CREATE TABLE foo (bar INTEGER);")
expect_equal(
driver$tableExists("foo"),
TRUE
)
expect_equal(
driver$tableExists("non-existent-table"),
FALSE
)
})
test_that("columnTypes", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update(paste0("DROP TABLE IF EXISTS ", driver$schema, ".foo CASCADE;"))
driver$update(paste0("CREATE TABLE ", driver$schema, ".foo (bar INTEGER, baz CHARACTER VARYING (200));"))
colTypes <- vector()
colTypes[["bar"]] <- "integer"
colTypes[["baz"]] <- "character varying"
expect_equal(
sort(driver$columnTypes('foo')),
sort(colTypes)
)
expect_equal(
length(driver$columnTypes("non-existent-table")),
0
)
})
test_that("saveDataFrame1", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame("foo" = c(1,3,5), "bar" = c("one", "three", "five"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
expect_equal(
dfResult,
df
)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame("foo" = c(1,3,5), "bar" = c("one", "three", "five"))
driver$saveDataFrame(df, "fooBar", rowNumbers = TRUE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar, row_num FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
df[['row_num']] <- c(1, 2, 3)
expect_equal(
dfResult,
df
)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
driver$update(paste0("CREATE TABLE ", RS_SCHEMA, ".fooBar (bar INTEGER);"))
# verify that the old table will get deleted
df <- data.frame("foo" = c(1,3,5), "bar" = c("one", "three", "five"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
expect_equal(
dfResult,
df
)
})
test_that("saveDataFrame2", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
driver$update(paste0("CREATE TABLE ", RS_SCHEMA, ".fooBar (bar INTEGER);"))
driver$update(paste0("CREATE VIEW ", RS_SCHEMA, ".basBar AS (SELECT * FROM ", RS_SCHEMA, ".fooBar);"))
# verify that the old table will get deleted even whant it has dependencies
df <- data.frame("foo" = c(1,3,5), "bar" = c("one", "three", "five"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
expect_equal(
dfResult,
df
)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
driver$update(paste0("CREATE TABLE ", RS_SCHEMA, ".fooBar (bar INTEGER);"))
# verify that the old table will not get deleted
expect_that(
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = TRUE),
throws_error()
)
df <- data.frame(name = c('first', 'second'))
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
driver$update(paste0("CREATE TABLE ", RS_SCHEMA, ".fooBar (name VARCHAR(200));"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = TRUE)
dfResult <- driver$select("SELECT name FROM fooBar ORDER BY name;")
dfResult[['name']] <- as.factor(dfResult[['name']])
expect_equal(
df,
dfResult
)
})
test_that("saveSingleRow", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame("foo" = c(1), "bar" = c("one"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
expect_equal(
dfResult,
df
)
})
test_that("saveDataFrameFile", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- read.table(header=TRUE, sep=",", text = '"","timestamp","anoms","expected_value"
"1","2011-02-25",9010345,9010345
"2","2013-08-25",747819,606634
"3","2013-09-20",779278,670728
"4","2013-10-04",827332,550889
"5","2013-10-06",788162,396039
"6","2013-10-11",1047828,198530
"7","2013-10-14",790715,1060
"8","2010-12-14",9002068,9002068
"9","2013-10-09",841979,442362
"10","2010-12-09",9003269,9003269
"11","2013-09-18",815260,728237
"12","2013-09-21",820278,743671
"13","2013-09-28",745668,772964
"14","2013-10-02",803688,803687
"15","2013-10-05",805601,669754
"16","2013-10-10",904069,535841
"17","2013-10-13",954868,408952
"18","2013-10-15",899812,282103
"19","2011-02-22",9011012,4857
"20","2013-05-31",952177,472330
"21","2013-09-15",751595,746646
"22","2013-09-19",743568,746286
"23","2013-10-03",802333,739189
"24","2013-10-12",1044813,733749')
df$timestamp <- as.POSIXlt(df$timestamp, tz = 'UTC')
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT \"timestamp\", anoms, expected_value FROM fooBar ORDER BY \"timestamp\";")
dfResult$timestamp <- as.POSIXlt(df$timestamp, tz = 'UTC')
expect_equal(nrow(df), nrow(df[which(dfResult$timestamp == df$timestamp),]))
})
test_that("saveDataFrameScientificNA", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(
id = c(1, 2, 6e+05),
text = character(3),
fact = c('fact1', 'fact2', 'fact2'),
stringsAsFactors = FALSE
)
df[1, 'text'] <- 'a'
df[2, 'text'] <- NA
df[['fact']] <- factor(df[['fact']])
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, forcedColumnTypes = list(id = "integer", text = "character"))
dfResult <- driver$select("SELECT id, text FROM fooBar;")
expect_equal(nrow(df), nrow(dfResult))
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(
id = c(1, 2, 6e+05, NA),
fact = c(12, NA, NA, 3),
stringsAsFactors = FALSE
)
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, forcedColumnTypes = list(id = "integer", fact = "character"))
dfResult <- driver$select("SELECT id, fact FROM fooBar;")
expect_equal(nrow(df), nrow(dfResult))
})
test_that("saveDataFrameLarge", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(a = rep('a', 10000), b = seq(1, 10000))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, displayProgress = FALSE)
dfff <- driver$select("SELECT a,b FROM fooBar ORDER BY b;")
dfResult <- driver$select("SELECT COUNT(*) AS cnt FROM fooBar;")
expect_equal(dfResult[1, 'cnt'], nrow(df))
})
test_that("saveDataFrameEscape", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(a = c('a', 'b'), b = c('foo \' bar', 'foo ; bar'))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, displayProgress = TRUE)
dfff <- driver$select("SELECT a,b FROM fooBar ORDER BY b;")
dfResult <- driver$select("SELECT COUNT(*) AS cnt FROM fooBar;")
expect_equal(dfResult[1, 'cnt'], nrow(df))
})
test_that("saveDataFrameNonScalar1", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(a = c('a', 'b'), stringsAsFactors = FALSE)
df$b <- list('e', 'f')
expect_error(
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, displayProgress = TRUE),
"non-scalar column"
)
})
test_that("saveDataFrameNonScalar2", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(a = c('a', 'b'), stringsAsFactors = FALSE)
df$b <- list(c('a1', 'a2', 'a3'), c('b1', 'b2'))
expect_error(
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, displayProgress = TRUE),
"non-scalar column"
)
})
keboola/backend-r-client documentation built on May 20, 2019, 8:29 a.m.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
library('keboola.backend.r.client')
test_that("connectRedshift", {
driver <- BackendDriver$new()
expect_equal(
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA),
TRUE
)
expect_that(
driver$connectRedshift("invalid", RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA),
throws_error()
)
})
test_that("prepare", {
driver <- BackendDriver$new()
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ?", "baz"),
"SELECT * FROM foo WHERE bar = 'baz'"
)
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ?", "ba'z"),
"SELECT * FROM foo WHERE bar = 'ba''z'"
)
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ?", 42),
"SELECT * FROM foo WHERE bar = '42'"
)
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ? AND baz = ?", list(bar=42,baz=21)),
"SELECT * FROM foo WHERE bar = '42' AND baz = '21'"
)
expect_equal(
driver$prepareStatement("SELECT * FROM foo WHERE bar = ? AND baz = ?", 42, 21),
"SELECT * FROM foo WHERE bar = '42' AND baz = '21'"
)
})
test_that("update", {
driver <- BackendDriver$new()
expect_that(
driver$update("CREATE TABLE foo (bar INTEGER);"),
throws_error()
)
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS foo CASCADE;")
expect_equal(
driver$update("CREATE TABLE foo (bar INTEGER);"),
TRUE
)
expect_that(
driver$update("CREATE TABLE foobar error;"),
throws_error()
)
})
test_that("update", {
driver <- BackendDriver$new()
expect_that(
driver$update("CREATE TABLE foo (bar INTEGER);"),
throws_error()
)
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS foo CASCADE;")
expect_equal(
driver$update("CREATE TABLE foo (bar INTEGER);"),
TRUE
)
expect_that(
driver$update("CREATE TABLE foobar error;"),
throws_error()
)
})
test_that("tableExists", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS foo CASCADE;")
driver$update("CREATE TABLE foo (bar INTEGER);")
expect_equal(
driver$tableExists("foo"),
TRUE
)
expect_equal(
driver$tableExists("non-existent-table"),
FALSE
)
})
test_that("columnTypes", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update(paste0("DROP TABLE IF EXISTS ", driver$schema, ".foo CASCADE;"))
driver$update(paste0("CREATE TABLE ", driver$schema, ".foo (bar INTEGER, baz CHARACTER VARYING (200));"))
colTypes <- vector()
colTypes[["bar"]] <- "integer"
colTypes[["baz"]] <- "character varying"
expect_equal(
sort(driver$columnTypes('foo')),
sort(colTypes)
)
expect_equal(
length(driver$columnTypes("non-existent-table")),
0
)
})
test_that("saveDataFrame1", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame("foo" = c(1,3,5), "bar" = c("one", "three", "five"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
expect_equal(
dfResult,
df
)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame("foo" = c(1,3,5), "bar" = c("one", "three", "five"))
driver$saveDataFrame(df, "fooBar", rowNumbers = TRUE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar, row_num FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
df[['row_num']] <- c(1, 2, 3)
expect_equal(
dfResult,
df
)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
driver$update(paste0("CREATE TABLE ", RS_SCHEMA, ".fooBar (bar INTEGER);"))
# verify that the old table will get deleted
df <- data.frame("foo" = c(1,3,5), "bar" = c("one", "three", "five"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
expect_equal(
dfResult,
df
)
})
test_that("saveDataFrame2", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
driver$update(paste0("CREATE TABLE ", RS_SCHEMA, ".fooBar (bar INTEGER);"))
driver$update(paste0("CREATE VIEW ", RS_SCHEMA, ".basBar AS (SELECT * FROM ", RS_SCHEMA, ".fooBar);"))
# verify that the old table will get deleted even whant it has dependencies
df <- data.frame("foo" = c(1,3,5), "bar" = c("one", "three", "five"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
expect_equal(
dfResult,
df
)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
driver$update(paste0("CREATE TABLE ", RS_SCHEMA, ".fooBar (bar INTEGER);"))
# verify that the old table will not get deleted
expect_that(
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = TRUE),
throws_error()
)
df <- data.frame(name = c('first', 'second'))
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
driver$update(paste0("CREATE TABLE ", RS_SCHEMA, ".fooBar (name VARCHAR(200));"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = TRUE)
dfResult <- driver$select("SELECT name FROM fooBar ORDER BY name;")
dfResult[['name']] <- as.factor(dfResult[['name']])
expect_equal(
df,
dfResult
)
})
test_that("saveSingleRow", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame("foo" = c(1), "bar" = c("one"))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT foo, bar FROM fooBar ORDER BY foo")
df[, "bar"] <- as.character(df[, "bar"])
expect_equal(
dfResult,
df
)
})
test_that("saveDataFrameFile", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- read.table(header=TRUE, sep=",", text = '"","timestamp","anoms","expected_value"
"1","2011-02-25",9010345,9010345
"2","2013-08-25",747819,606634
"3","2013-09-20",779278,670728
"4","2013-10-04",827332,550889
"5","2013-10-06",788162,396039
"6","2013-10-11",1047828,198530
"7","2013-10-14",790715,1060
"8","2010-12-14",9002068,9002068
"9","2013-10-09",841979,442362
"10","2010-12-09",9003269,9003269
"11","2013-09-18",815260,728237
"12","2013-09-21",820278,743671
"13","2013-09-28",745668,772964
"14","2013-10-02",803688,803687
"15","2013-10-05",805601,669754
"16","2013-10-10",904069,535841
"17","2013-10-13",954868,408952
"18","2013-10-15",899812,282103
"19","2011-02-22",9011012,4857
"20","2013-05-31",952177,472330
"21","2013-09-15",751595,746646
"22","2013-09-19",743568,746286
"23","2013-10-03",802333,739189
"24","2013-10-12",1044813,733749')
df$timestamp <- as.POSIXlt(df$timestamp, tz = 'UTC')
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE)
dfResult <- driver$select("SELECT \"timestamp\", anoms, expected_value FROM fooBar ORDER BY \"timestamp\";")
dfResult$timestamp <- as.POSIXlt(df$timestamp, tz = 'UTC')
expect_equal(nrow(df), nrow(df[which(dfResult$timestamp == df$timestamp),]))
})
test_that("saveDataFrameScientificNA", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(
id = c(1, 2, 6e+05),
text = character(3),
fact = c('fact1', 'fact2', 'fact2'),
stringsAsFactors = FALSE
)
df[1, 'text'] <- 'a'
df[2, 'text'] <- NA
df[['fact']] <- factor(df[['fact']])
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, forcedColumnTypes = list(id = "integer", text = "character"))
dfResult <- driver$select("SELECT id, text FROM fooBar;")
expect_equal(nrow(df), nrow(dfResult))
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(
id = c(1, 2, 6e+05, NA),
fact = c(12, NA, NA, 3),
stringsAsFactors = FALSE
)
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, forcedColumnTypes = list(id = "integer", fact = "character"))
dfResult <- driver$select("SELECT id, fact FROM fooBar;")
expect_equal(nrow(df), nrow(dfResult))
})
test_that("saveDataFrameLarge", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(a = rep('a', 10000), b = seq(1, 10000))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, displayProgress = FALSE)
dfff <- driver$select("SELECT a,b FROM fooBar ORDER BY b;")
dfResult <- driver$select("SELECT COUNT(*) AS cnt FROM fooBar;")
expect_equal(dfResult[1, 'cnt'], nrow(df))
})
test_that("saveDataFrameEscape", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(a = c('a', 'b'), b = c('foo \' bar', 'foo ; bar'))
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, displayProgress = TRUE)
dfff <- driver$select("SELECT a,b FROM fooBar ORDER BY b;")
dfResult <- driver$select("SELECT COUNT(*) AS cnt FROM fooBar;")
expect_equal(dfResult[1, 'cnt'], nrow(df))
})
test_that("saveDataFrameNonScalar1", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(a = c('a', 'b'), stringsAsFactors = FALSE)
df$b <- list('e', 'f')
expect_error(
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, displayProgress = TRUE),
"non-scalar column"
)
})
test_that("saveDataFrameNonScalar2", {
driver <- BackendDriver$new()
driver$connectRedshift(RS_HOST, RS_DB, RS_USER, RS_PASSWORD, RS_SCHEMA)
driver$update("DROP TABLE IF EXISTS fooBar CASCADE;")
df <- data.frame(a = c('a', 'b'), stringsAsFactors = FALSE)
df$b <- list(c('a1', 'a2', 'a3'), c('b1', 'b2'))
expect_error(
driver$saveDataFrame(df, "fooBar", rowNumbers = FALSE, incremental = FALSE, displayProgress = TRUE),
"non-scalar column"
)
})
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Embedding an R snippet on your website
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.