Fix can_offset loading and tests

sdss · Apr 18, 2024 · 3a97228 · 3a97228
1 parent cf79627
commit 3a97228
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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');