data("nba_clusts")

test_that("node_colour aesthetic works", {
    expect_is(clustree(nba_clusts, prefix = "K", node_colour = "ReboundPct",
                       node_colour_aggr = "mean"),
              c("gg", "ggplot"))
    expect_error(clustree(nba_clusts, prefix = "K",
                          node_colour = "ReboundPct"),
              "Must be of type 'character'")
    expect_error(clustree(nba_clusts, prefix = "K",
                          node_colour = "ReboundPct",
                          node_colour_aggr = "FAKEFUN"),
                 "'function' was not found")
})

test_that("static node_colour works", {
    expect_is(clustree(nba_clusts, prefix = "K", node_colour = 1),
              c("gg", "ggplot"))
    expect_is(clustree(nba_clusts, prefix = "K", node_colour = "purple"),
              c("gg", "ggplot"))
    expect_error(clustree(nba_clusts, prefix = "K", node_colour = -1),
                 ">= 0")
    expect_error(clustree(nba_clusts, prefix = "K", node_colour = "XXXX"),
                 "not a valid colour name")
})

test_that("node_size aesthetic works", {
    expect_is(clustree(nba_clusts, prefix = "K", node_size = "ReboundPct",
                       node_size_aggr = "mean"),
              c("gg", "ggplot"))
    expect_error(clustree(nba_clusts, prefix = "K",
                          node_size = "ReboundPct"),
                 "Must be of type 'character'")
    expect_error(clustree(nba_clusts, prefix = "K",
                          node_size = "ReboundPct",
                          node_size_aggr = "FAKEFUN"),
                 "'function' was not found")
})

test_that("static node_size works", {
    expect_is(clustree(nba_clusts, prefix = "K", node_size = 1),
              c("gg", "ggplot"))
    expect_error(clustree(nba_clusts, prefix = "K", node_size = -1),
                 ">= 0")
    expect_error(clustree(nba_clusts, prefix = "K", node_size = "XXXX"),
                 "must be one of")
})

test_that("node_alpha aesthetic works", {
    expect_is(clustree(nba_clusts, prefix = "K", node_alpha = "ReboundPct",
                       node_alpha_aggr = "mean"),
              c("gg", "ggplot"))
    expect_error(clustree(nba_clusts, prefix = "K",
                          node_alpha = "ReboundPct"),
                 "Must be of type 'character'")
    expect_error(clustree(nba_clusts, prefix = "K",
                          node_alpha = "ReboundPct",
                          node_alpha_aggr = "FAKEFUN"),
                 "'function' was not found")
})

test_that("static node_size works", {
    expect_is(clustree(nba_clusts, prefix = "K", node_alpha = 1),
              c("gg", "ggplot"))
    expect_error(clustree(nba_clusts, prefix = "K", node_alpha = -1),
                 ">= 0")
    expect_error(clustree(nba_clusts, prefix = "K", node_alpha = 2),
                 "<= 1")
    expect_error(clustree(nba_clusts, prefix = "K", node_alpha = "XXXX"),
                 "must be one of")
})

test_that("all static works", {
    expect_is(clustree(nba_clusts, prefix = "K", node_colour = "red",
                       node_size = 10, node_alpha = 1),
              c("gg", "ggplot"))
})

test_that("layout argument works", {
    expect_is(clustree(nba_clusts, prefix = "K", layout = "sugiyama"),
              c("gg", "ggplot"))
    expect_error(clustree(nba_clusts, prefix = "K", layout = "apple"),
                 "should be one of")
})

test_that("Colour, Size, Alpha node aesthetics work", {
    expect_is(
        clustree(nba_clusts, prefix = "K",
                 node_size = "size", node_colour = "size", node_alpha = "size"),
        c("gg", "ggplot")
    )
})

test_that("Colour, Alpha node aesthetics work", {
    expect_is(
        clustree(nba_clusts, prefix = "K",
                 node_size = 1, node_colour = "size", node_alpha = "size"),
        c("gg", "ggplot")
    )
})

test_that("Colour node aesthetics work", {
    expect_is(
        clustree(nba_clusts, prefix = "K",
                 node_size = 1, node_colour = "size", node_alpha = 1),
        c("gg", "ggplot")
    )
})

test_that("Size, alpha node aesthetics work", {
    expect_is(
        clustree(nba_clusts, prefix = "K",
                 node_size = "size", node_colour = 1, node_alpha = "size"),
        c("gg", "ggplot")
    )
})

test_that("Size node aesthetics work", {
    expect_is(
        clustree(nba_clusts, prefix = "K",
                 node_size = "size", node_colour = 1, node_alpha = 1),
        c("gg", "ggplot")
    )
})

test_that("Alpha node aesthetics work", {
    expect_is(
        clustree(nba_clusts, prefix = "K",
                 node_size = 1, node_colour = 1, node_alpha = "size"),
        c("gg", "ggplot")
    )
})

test_that("None node aesthetics work", {
    expect_is(
        clustree(nba_clusts, prefix = "K",
                 node_size = 1, node_colour = 1, node_alpha = 1),
        c("gg", "ggplot")
    )
})