# agridatasets - A Comprehensive Collection of Agricultural and Agronomic Datasets
# Version 0.1.1
# Copyright (C) 2026 Renzo Caceres Rossi
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

#   pollen_removal

library(testthat)

# Test 1: Check that the object is a data.frame and has 3 columns
test_that("pollen_removal is a data.frame with 3 columns", {
  expect_s3_class(pollen_removal, "data.frame")
  expect_equal(ncol(pollen_removal), 3)
  expect_equal(length(pollen_removal), 3)
})

# Test 2: Check the number of rows
test_that("pollen_removal has 47 rows", {
  expect_equal(nrow(pollen_removal), 47)
})

# Test 3: Validate column names
test_that("pollen_removal has correct column names", {
  expect_named(
    pollen_removal,
    c("PollenRemoved", "DurationOfVisit", "BeeType")
  )
})

# Test 4: Check data types
test_that("pollen_removal has correct data types", {
  expect_type(pollen_removal$PollenRemoved, "double")
  expect_type(pollen_removal$DurationOfVisit, "integer")
  expect_s3_class(pollen_removal$BeeType, "factor")
  expect_equal(nlevels(pollen_removal$BeeType), 2)
})