diff --git a/tests/unit/test_wildlife_schema.py b/tests/unit/test_wildlife_schema.py new file mode 100644 index 0000000..61fd434 --- /dev/null +++ b/tests/unit/test_wildlife_schema.py @@ -0,0 +1,13 @@ +from vigilar.storage.schema import wildlife_sightings + + +def test_wildlife_sightings_has_temperature(): + assert "temperature_c" in [c.name for c in wildlife_sightings.columns] + + +def test_wildlife_sightings_has_conditions(): + assert "conditions" in [c.name for c in wildlife_sightings.columns] + + +def test_wildlife_sightings_has_bbox(): + assert "bbox" in [c.name for c in wildlife_sightings.columns] diff --git a/vigilar/storage/schema.py b/vigilar/storage/schema.py index 410af24..d083ba1 100644 --- a/vigilar/storage/schema.py +++ b/vigilar/storage/schema.py @@ -165,6 +165,9 @@ wildlife_sightings = Table( Column("confidence", Float), Column("crop_path", String), Column("event_id", Integer), + Column("temperature_c", Float), + Column("conditions", String), + Column("bbox", String), # JSON [x, y, w, h] normalized ) Index("idx_wildlife_ts", wildlife_sightings.c.ts.desc()) Index("idx_wildlife_threat", wildlife_sightings.c.threat_level, wildlife_sightings.c.ts.desc())