codelist class

DESCRIPTION

@@ -12,7 +12,7 @@ Description: Generate a candidate code list for the Observational Medical Outcom
 License: Apache License (>= 2)
 Encoding: UTF-8
 Roxygen: list(markdown = TRUE)
-RoxygenNote: 7.2.3
+RoxygenNote: 7.3.1
 Depends:
     R (>= 3.5.0)
 Imports:

R/drugCodes.R

@@ -144,6 +144,7 @@ getATCCodes <- function(cdm,
         dplyr::select("concept_id") %>%
         dplyr::distinct() %>%
         dplyr::pull()
+
       } else {
         atc_descendants[[i]] <- atc_descendants[[i]] %>%
           dplyr::select(!"ancestor_concept_id")
@@ -152,6 +153,11 @@ getATCCodes <- function(cdm,
       names(atc_descendants)[i] <- workingName
     }
   }
+
+  if(isFALSE(withConceptDetails)){
+  atc_descendants <- omopgenerics::newCodelist(atc_descendants)
+  }
+
   return(atc_descendants)
 }
 
@@ -278,13 +284,19 @@ getDrugIngredientCodes <- function(cdm,
           dplyr::select("concept_id") %>%
           dplyr::distinct() %>%
           dplyr::pull()
+
       } else {
         ingredientCodes[[i]] <- ingredientCodes[[i]] %>%
           dplyr::select(!"ancestor_concept_id")
       }
 
       names(ingredientCodes)[[i]] <- workingName
     }
+
+    if(isFALSE(withConceptDetails)){
+    ingredientCodes <- omopgenerics::newCodelist(ingredientCodes)
+    }
+
     return(ingredientCodes)
 }
 

R/getICD10StandardCodes.R

@@ -141,6 +141,11 @@ getICD10StandardCodes <- function(cdm,
     )
   }
 
+
+  if(isFALSE(withConceptDetails)){
+    ICD10StandardCodes <- omopgenerics::newCodelist(ICD10StandardCodes)
+  }
+
   return(ICD10StandardCodes)
 }
 

tests/testthat/test-drugCodes.R

@@ -4,6 +4,7 @@ test_that("getATCCodes working", {
   for (i in seq_along(backends)) {
   cdm <- mockVocabRef(backend = backends[i])
   atcCodes <- getATCCodes(cdm, level = "ATC 1st")
+  expect_true("codelist" %in% class(atcCodes))
   expect_true(all(atcCodes[[1]] == c(12,13)))
   expect_true(c("alimentary_tract_and_metabolism") %in%
                 names(atcCodes))