Add int tests for no_gaps

veridit · Nov 6, 2023 · 79ec9f3 · 79ec9f3
1 parent 507f2a4
commit 79ec9f3
Show file tree

Hide file tree

Showing 2 changed files with 215 additions and 0 deletions.
diff --git a/expected/31_no_gaps__int_test.out b/expected/31_no_gaps__int_test.out
@@ -0,0 +1,135 @@
+-- First, create the integer shifts table
+CREATE TABLE int_shifts (
+  job_id INTEGER,
+  worker_id INTEGER,
+  valid_from INT,
+  valid_to INT
+);
+-- Add era and unique key via sql_saga
+SELECT sql_saga.add_era('int_shifts', 'valid_from', 'valid_to');
+ add_era 
+---------
+ t
+(1 row)
+
+SELECT sql_saga.add_unique_key('int_shifts', ARRAY['job_id', 'worker_id']);
+          add_unique_key           
+-----------------------------------
+ int_shifts_job_id_worker_id_valid
+(1 row)
+
+TABLE sql_saga.era;
+ table_name | era_name | start_column_name | end_column_name | range_type | bounds_check_constraint | audit_table_name 
+------------+----------+-------------------+-----------------+------------+-------------------------+------------------
+ int_shifts | valid    | valid_from        | valid_to        | int4range  | int_shifts_valid_check  | 
+(1 row)
+
+TABLE sql_saga.unique_keys;
+             key_name              | table_name |    column_names    | era_name |                  unique_constraint                  |             exclude_constraint             
+-----------------------------------+------------+--------------------+----------+-----------------------------------------------------+--------------------------------------------
+ int_shifts_job_id_worker_id_valid | int_shifts | {job_id,worker_id} | valid    | int_shifts_job_id_worker_id_valid_from_valid_to_key | int_shifts_job_id_worker_id_int4range_excl
+(1 row)
+
+-- Insert test data into the integer shifts table
+INSERT INTO int_shifts(job_id, worker_id, valid_from, valid_to) VALUES
+  (1, 1, 1, 6),
+  (1, 2, 6, 12),
+  (2, 3, 1, 6),
+  (2, 4, 7, 12),
+  (3, 5, 0, 6),  -- Using 0 to indicate the start of our range
+  (3, 5, 6, 12),
+  (4, 6, 1, 6),
+  (4, 7, 6, 2147483647);  -- Using the max integer to indicate 'infinity'
+-- This test checks for an exact match with one range
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(1, 6))
+FROM int_shifts
+WHERE job_id = 1;
+ no_gaps 
+---------
+ t
+(1 row)
+
+-- This test checks for an exact match with two consecutive ranges
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(1, 12))
+FROM int_shifts
+WHERE job_id = 1;
+ no_gaps 
+---------
+ t
+(1 row)
+
+-- Test 3: Range with Extra at the Beginning
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(2, 6))
+FROM int_shifts
+WHERE job_id = 1;
+ no_gaps 
+---------
+ t
+(1 row)
+
+-- Test 4: Range with Extra at the End
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(1, 11))
+FROM int_shifts
+WHERE job_id = 1;
+ no_gaps 
+---------
+ t
+(1 row)
+
+-- Test 5: Range with Extra on Both Sides
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(2, 11))
+FROM int_shifts
+WHERE job_id = 1;
+ no_gaps 
+---------
+ t
+(1 row)
+
+-- Test 6: Range that Misses Completely
+-- Expected: FALSE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(20, 25))
+FROM int_shifts
+WHERE job_id = 1;
+ no_gaps 
+---------
+ f
+(1 row)
+
+-- Test 7: Range with Uncovered Time at the Beginning
+-- Expected: FALSE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(0, 6))
+FROM int_shifts
+WHERE job_id = 1;
+ no_gaps 
+---------
+ f
+(1 row)
+
+-- Test 8: Range with Uncovered Time at the End
+-- Expected: FALSE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(1, 15))
+FROM int_shifts
+WHERE job_id = 1;
+ no_gaps 
+---------
+ f
+(1 row)
+
+SELECT sql_saga.drop_unique_key('int_shifts', 'int_shifts_job_id_worker_id_valid');
+ drop_unique_key 
+-----------------
+
+(1 row)
+
+SELECT sql_saga.drop_era('int_shifts');
+ drop_era 
+----------
+ t
+(1 row)
+
+DROP TABLE int_shifts;
diff --git a/sql/31_no_gaps__int_test.sql b/sql/31_no_gaps__int_test.sql
@@ -0,0 +1,80 @@
+-- First, create the integer shifts table
+CREATE TABLE int_shifts (
+  job_id INTEGER,
+  worker_id INTEGER,
+  valid_from INT,
+  valid_to INT
+);
+-- Add era and unique key via sql_saga
+SELECT sql_saga.add_era('int_shifts', 'valid_from', 'valid_to');
+SELECT sql_saga.add_unique_key('int_shifts', ARRAY['job_id', 'worker_id']);
+
+TABLE sql_saga.era;
+TABLE sql_saga.unique_keys;
+
+
+-- Insert test data into the integer shifts table
+INSERT INTO int_shifts(job_id, worker_id, valid_from, valid_to) VALUES
+  (1, 1, 1, 6),
+  (1, 2, 6, 12),
+  (2, 3, 1, 6),
+  (2, 4, 7, 12),
+  (3, 5, 0, 6),  -- Using 0 to indicate the start of our range
+  (3, 5, 6, 12),
+  (4, 6, 1, 6),
+  (4, 7, 6, 2147483647);  -- Using the max integer to indicate 'infinity'
+
+
+-- This test checks for an exact match with one range
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(1, 6))
+FROM int_shifts
+WHERE job_id = 1;
+
+-- This test checks for an exact match with two consecutive ranges
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(1, 12))
+FROM int_shifts
+WHERE job_id = 1;
+
+-- Test 3: Range with Extra at the Beginning
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(2, 6))
+FROM int_shifts
+WHERE job_id = 1;
+
+-- Test 4: Range with Extra at the End
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(1, 11))
+FROM int_shifts
+WHERE job_id = 1;
+
+-- Test 5: Range with Extra on Both Sides
+-- Expected: TRUE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(2, 11))
+FROM int_shifts
+WHERE job_id = 1;
+
+-- Test 6: Range that Misses Completely
+-- Expected: FALSE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(20, 25))
+FROM int_shifts
+WHERE job_id = 1;
+
+-- Test 7: Range with Uncovered Time at the Beginning
+-- Expected: FALSE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(0, 6))
+FROM int_shifts
+WHERE job_id = 1;
+
+-- Test 8: Range with Uncovered Time at the End
+-- Expected: FALSE
+SELECT sql_saga.no_gaps(int4range(valid_from, valid_to), int4range(1, 15))
+FROM int_shifts
+WHERE job_id = 1;
+
+
+SELECT sql_saga.drop_unique_key('int_shifts', 'int_shifts_job_id_worker_id_valid');
+SELECT sql_saga.drop_era('int_shifts');
+
+DROP TABLE int_shifts;