Intiaial dispatch for period_prior_to_index function

src/preprocessing.jl

@@ -26,4 +26,29 @@ function Dummy(
 
 end
 
-export Dummy
+
+"""
+TODO: Add Doc-Strings.
+"""
+function period_prior_to_index(cohort_id::Vector, conn; date_prior=Day(100), tab=cohort)
+
+ # Construct the SQL query
+ sql = From(tab) |>
+ Where(Fun.in(Get.cohort_definition_id, cohort_id...)) |>
+ Select(Get.cohort_definition_id, Get.subject_id, Get.cohort_start_date) |>
+ q -> render(q, dialect=dialect)
+
+ # Execute the SQL query and fetch the result into a DataFrame
+ df = DBInterface.execute(conn, String(sql)) |> DataFrame
+
+ # Check if the DataFrame is not empty
+ # TODO: Is it really necessary to add this check ??
+ if nrow(df) > 0
+ # Convert the cohort_start_date to DateTime and subtract the date_prior
+ df.cohort_start_date = DateTime.(df.cohort_start_date) .- date_prior
+ end
+
+ return df
+end
+
+export Dummy, period_prior_to_index

test/Data-Preprocessing/preprocessing.jl

@@ -10,4 +10,22 @@ using Test
  @test test_drug_exposure_start_date == result.drug_exposure_start_date[1:4]
  @test test_drug_exposure_ids == result.drug_exposure_id[1:4]
 
+end
+
+
+
+@testset "Period Prior to Index Tests" begin
+ MakeTables(sqlite_conn, :sqlite, "main")
+
+ test_person_ids = [1, 1, 1, 1, 1]
+ test_subject_ids = [1.0, 5.0, 9.0, 11.0, 12.0]
+ test_cohort_start_date = [-3.7273e8, 2.90304e7, -5.33347e8, -8.18208e7, 1.32918e9]
+
+ test_df2 = DataFrame(person_id = test_person_ids, cohort_start_date = test_cohort_start_date)
+
+ result = period_prior_to_index(test_person_ids, sqlite_conn)
+
+ @test test_person_ids == result.cohort_definition_id[1:5]
+ @test test_subject_ids == result.subject_id[1:5]
+
 end