HASP: Hydrologic AnalySis Package

context("Visualize Data")

test_that("Composite Graphs", {
  aquifer_data <- aquifer_data
  num_years <- 30
 
  comp_data <- plot_composite_data(aquifer_data,  num_years, 
                                   parameter_cd = "72019")
  norm_data <- plot_normalized_data(aquifer_data, num_years, 
                                    parameter_cd = "72019")
  
  expect_true(all(names(comp_data$data) %in% c("year","name","value")))
  expect_true(all(names(norm_data$data) %in% c("year","name","value")))

  expect_true(all(levels(comp_data$data$name) %in% c("Median", "Mean")))
  expect_true(all(levels(norm_data$data$name) %in% c("Median",
                                                     "Mean")))
  
    
})
  
test_that("Map", {
  aquifer_data <- aquifer_data
  num_years <- 30
  
  map <- map_hydro_data(aquifer_data, num_years, parameter_cd = "72019")
  
  expect_true(all(class(map) %in% c("leaflet","htmlwidget")))

  
})


test_that("SC Chloride graphs and table", {

  site_data <- L2701_example_data$QW
  
  sc_plot <- Sc_Cl_plot(site_data, plot_title = "Hi")
  expect_true(all(c("Date", "Chloride", "Specific conductance") %in%
                    names(sc_plot$data)))
  
  expect_error(Sc_Cl_plot(site_data[FALSE,], plot_title = "Hi"))
  
  sc_plot <- Sc_Cl_plot(site_data, plot_title = "Hi", subtitle = "NWIS")
  expect_true(all(c("Date", "Chloride", "Specific conductance") %in%
                    names(sc_plot$data)))
  
  sccl_table <- Sc_Cl_table(site_data)
  expect_true(all(c("Date",   
                    "Chloride",
                    "Specific conductance" ) %in% names(sccl_table)))
  
  expect_error(Sc_Cl_table(site_data[,c("ActivityStartDateTime")]))
  
  qw_plot_out <- qw_plot(site_data, "hi!")
  expect_true(all(c(c("data", "layers", "scales",     
                      "mapping", "theme", "coordinates",
                      "facet","plot_env", "labels")) %in%
                    names(qw_plot_out)))
  
})

test_that("Monthly frequency plot", {
  
  plot <- monthly_frequency_plot(L2701_example_data$Daily,
                                 L2701_example_data$Discrete,
                                 parameter_cd = "62610")
  
  plot_data_elements <- unlist(lapply(plot$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("plot_month", "ymin", "ymax", "month", "value", "group") %in%
                plot_data_elements))
  
  plot <- monthly_frequency_plot(L2701_example_data$Daily,
                                 NULL,
                                 parameter_cd = "62610")
  
  plot_data_elements <- unlist(lapply(plot$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("plot_month", "ymin", "ymax", "month", "value", "group") %in%
                    plot_data_elements))
  
  plot <- monthly_frequency_plot(L2701_example_data$Daily,
                                 L2701_example_data$Discrete,
                                 parameter_cd = "62610",
                                 include_edges = TRUE)
  
  plot_data_elements <- unlist(lapply(plot$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("plot_month", "ymin", "ymax", "month", "value", "group") %in%
                    plot_data_elements))
  
  plot <- monthly_frequency_plot(L2701_example_data$Daily,
                                 L2701_example_data$Discrete,
                                 parameter_cd = "62610",
                                 percentile_colors = c("#d7191c", "#fdae61", "#ffffbf", "#abdda4", "#2b83ba"))
  
  plot_data_elements <- unlist(lapply(plot$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("plot_month", "ymin", "ymax", "month", "value", "group") %in%
                    plot_data_elements))
  
})

test_that("Weekly frequency plot", {
  
  plot <- weekly_frequency_plot(L2701_example_data$Daily,
                                L2701_example_data$Discrete,
                                parameter_cd = "62610")
  
  plot_data_elements <- unlist(lapply(plot$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("plot_week", "x", "y", "ymin", "group", "plot_week_last") %in%
                    plot_data_elements))
  
  plot <- weekly_frequency_plot(L2701_example_data$Daily,
                                L2701_example_data$Discrete,
                                date_col = c("Date", "lev_dt"),
                                value_col = c("X_62610_00001", "sl_lev_va"),
                                approved_col = c("X_62610_00001_cd", "lev_status_cd"))
  
})

test_that("Field gwl plot", {
  
  gwl_data <- L2701_example_data$Discrete
  plot_title <- attr(gwl_data, "siteInfo")[["station_nm"]]
  plot_out <- gwl_plot_field(gwl_data, plot_title, parameter_cd = "62610")
  
  dv <- L2701_example_data$Daily
  plot2 <- gwl_plot_all(dv, gwl_data,
                        parameter_cd = "62610",
                        plot_title = "title")

  expect_true(all(c("data", "layers", "scales",     
                    "mapping", "theme", "coordinates",
                    "facet","plot_env", "labels") %in%
                    names(plot2)))
  
  plot_with_trend <- gwl_plot_all(dv, gwl_data, 
                                  plot_title = "title",
                                  parameter_cd = "62610",
                                  add_trend = TRUE)
  
  plot_data_elements <- unlist(lapply(plot_with_trend$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("Date", "Value", "year", 
                    "is_complete", "x1", "x2", "y1", "y2", "trend") %in%
                    plot_data_elements))
  
  plot_with_gwl <- gwl_plot_all(NULL, gwl_data, 
                                plot_title = "title",
                                parameter_cd = "62610",
                                add_trend = FALSE)
  plot_elements <- unlist(lapply(plot_with_gwl$layers, function(x) {names(x$data)}))
  expect_true(all(c("Date", "Value", "year", "Approve") %in%
                    plot_elements))
  
})

test_that("Daily gwl plot", {
  
  gw_daily <- L2701_example_data$Daily
  gw_discrete <- L2701_example_data$Discrete
  
  plot_title <- attr(gw_daily, "siteInfo")[["station_nm"]]
  
  plot1 <- daily_gwl_plot(gw_daily, 
                          NULL,
                          date_col = "Date",
                          value_col = "X_62610_00001",
                          approved_col = "X_62610_00001_cd",
                          historical_stat = "mean",
                          month_breaks = TRUE,
                          plot_title = plot_title)
  
  plot_data_elements <- unlist(lapply(plot1$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("Date", "middle", "min", "max") %in%
                    plot_data_elements))
  
  plot2 <- daily_gwl_plot(gw_daily, 
                          gw_discrete,
                          parameter_cd = "62610",
                          stat_cd = "00001",
                          historical_stat = "median",
                          month_breaks = FALSE,
                          plot_title = plot_title)
  
  plot_data_elements <- unlist(lapply(plot2$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("Date", "middle", "min", "max") %in%
                    plot_data_elements))
  
  plot3 <- daily_gwl_plot(gw_daily, NULL,
                              parameter_cd = "62610",
                              historical_stat = "median",
                              month_breaks = FALSE,
                              plot_title = plot_title)
  
  plot_data_elements <- unlist(lapply(plot3$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("Date", "middle", "min", "max") %in%
                    plot_data_elements))
  
  plot4 <- daily_gwl_plot(gw_daily, 
                          gw_discrete,
                          date_col = c("Date",
                                       "lev_dt"),
                          value_col = c("X_62610_00001",
                                        "sl_lev_va"),
                          approved_col = c("X_62610_00001_cd",
                                           "lev_status_cd"),
                          start_date = "2018-10-01",
                          end_date = "2020-10-01",
                          historical_stat = "mean",
                          month_breaks = TRUE,
                          plot_title = plot_title)
  
  plot_data_elements <- unlist(lapply(plot4$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("Date", "middle", "min", "max") %in%
                    plot_data_elements))
  
})

test_that("Chloride trend graph", {
  qw_data <- L2701_example_data$QW
  title <- "Hi"
  plot_out <- trend_plot(qw_data, plot_title = title)
  plot_data_elements <- unlist(lapply(plot_out$layers, function(x) {names(x$data)}))
  
  expect_true(all(c("ActivityStartDate", "ResultMeasureValue", "condition", "x1", "x2", "y1",       
                    "y2", "trend", "y") %in% plot_data_elements))
})