Add test for sqlite (#69)

* Test if database is generated

* All tests are OK

* Add logic to remove existing database

* Bump version

Author: vnegi10

Project.toml

@@ -1,7 +1,7 @@
 name = "WeatherReport"
 uuid = "3c954167-71e4-4212-9688-830d6b8fcbc2"
 authors = ["Vikas Negi <vikas.negi10@gmail.com>"]
-version = "0.18.0"
+version = "0.18.1"
 
 [deps]
 CSV = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b"

test/Project.toml

@@ -1,4 +1,6 @@
 [deps]
+DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
 ReTestItems = "817f1d60-ba6b-4fd5-9520-3cf149f6a823"
+SQLite = "0aa819cd-b072-5ff4-a722-6bc24af294d9"
 Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

test/export_history_test.jl

@@ -0,0 +1,68 @@
+# Run tests on exporting historical data to SQLite database
+
+@testsetup module TestDatabase
+
+    export db, df_snow, df_rain, df_temp
+
+    using WeatherReport, SQLite, DataFrames
+
+    db_path = joinpath(@__DIR__, "..", "export", "Veldhoven_all.sqlite")
+
+    # Remove existing database otherwise new data gets appended to it
+    if isfile(db_path)
+        rm(db_path)
+    end
+
+    export_to_sqlite("Veldhoven",
+                      start_date = "2022-01-01",
+                      end_date = "2022-01-31")
+    
+    db = SQLite.DB(db_path)
+
+    df_snow = DBInterface.execute(db, "SELECT * FROM snowfall") |> DataFrame
+    df_rain = DBInterface.execute(db, "SELECT * FROM rain") |> DataFrame
+    df_temp = DBInterface.execute(db, "SELECT * FROM temperature_2m") |> DataFrame
+end
+
+@testitem "SQLite database" setup = [TestDatabase] begin
+
+    using SQLite
+
+    @testset "Check number of tables" begin
+        @test SQLite.tables(db) |> length == 6
+    end
+
+    @testset "Check snowfall data size" begin
+        rows, cols = size(df_snow)
+
+        @test rows == 744
+        @test cols == 2
+    end
+
+    @testset "Check rain data" begin
+        df_filter = filter(row -> row.rain > 0, df_rain)
+        rows, cols = size(df_filter)
+
+        @test rows == 152
+        @test cols == 2
+
+        @test df_rain[!, :rain][2] ≈ 0.1
+        @test df_rain[!, :rain][742] ≈ 0.1
+
+    end
+
+    @testset "Check temperature data" begin
+        df_filter = filter(row -> row.temperature_2m > 5.5, df_temp)
+        rows, cols = size(df_filter)
+
+        @test rows == 208
+        @test cols == 3
+
+        @test df_temp[!, :temperature_2m][1] ≈ 12.3
+        @test df_temp[!, :temperature_2m][744] ≈ 3.5
+
+        @test df_temp[!, :APP_TEMP][2] ≈ 11.3
+        @test df_temp[!, :APP_TEMP][743] ≈ -0.6
+    end
+
+end