From 3a97228defcefede0d28953e12988b0af509c240 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20S=C3=A1nchez-Gallego?= <gallegoj@uw.edu>
Date: Thu, 18 Apr 2024 19:46:02 +0000
Subject: [PATCH] Fix can_offset loading and tests

---
 src/too/datamodel.py               | 1 +
 src/too/mock.py                    | 1 +
 tests/scripts/sdss5db_too_test.sql | 3 ++-
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/too/datamodel.py b/src/too/datamodel.py
index b767d13..24dd4f1 100644
--- a/src/too/datamodel.py
+++ b/src/too/datamodel.py
@@ -73,6 +73,7 @@
     "h_mag",
     "delta_ra",
     "delta_dec",
+    "can_offset",
     "inertial",
     "n_exposures",
     "priority",
diff --git a/src/too/mock.py b/src/too/mock.py
index 738ff11..227ada6 100644
--- a/src/too/mock.py
+++ b/src/too/mock.py
@@ -277,6 +277,7 @@ def create_mock_too_catalogue(
     df = df.with_columns(
         fiber_type=polars.Series(fiber_type, dtype=polars.String),
         observed=False,
+        can_offset=True,
         active=True,
         priority=polars.lit(5, dtype=polars.Int16),
         n_exposures=polars.lit(3, dtype=polars.Int16),
diff --git a/tests/scripts/sdss5db_too_test.sql b/tests/scripts/sdss5db_too_test.sql
index 3d899f3..be433a1 100644
--- a/tests/scripts/sdss5db_too_test.sql
+++ b/tests/scripts/sdss5db_too_test.sql
@@ -143,6 +143,7 @@ CREATE TABLE catalogdb.too_metadata (
     h_mag REAL,
     delta_ra REAL,
     delta_dec REAL,
+    can_offset BOOLEAN,
     inertial BOOLEAN,
     n_exposures SMALLINT,
     priority SMALLINT,
@@ -344,7 +345,7 @@ CREATE UNIQUE INDEX ON targetdb.cadence(label text_ops);
 CREATE UNIQUE INDEX ON targetdb.cadence(label text_ops);
 CREATE INDEX ON targetdb.cadence(nepochs int4_ops);
 
-CREATE UNIQUE INDEX design_mode_pkey ON targetdb.design_mode(label);
+CREATE UNIQUE INDEX ON targetdb.design_mode(label);
 
 INSERT INTO catalogdb.version VALUES (31, '1.0.0', '1.0.0');
 INSERT INTO targetdb.cadence VALUES ('bright_1x1', 1, '{0}', '{1}', '{0}', '{0}', '{1}', '{0}', 1, null, 'bright_1x1');