fix(right-wing): match store_scores column names and value order to DB schema

main
Sven Geboers 4 weeks ago
parent 84ec44e468
commit 80c68c0112
  1. 12
      analysis/right_wing/extremity_rescore_2d.py
  2. 4
      tests/right_wing/test_extremity_rescore_2d.py

@ -228,10 +228,10 @@ def store_scores(db_path: str, results: list[dict[str, Any]]) -> int:
"""
CREATE TABLE IF NOT EXISTS extremity_scores_2d (
motion_id INTEGER PRIMARY KEY,
stylistic_score INTEGER NOT NULL,
material_score INTEGER NOT NULL,
stylistic_rationale TEXT,
material_rationale TEXT
stijl_extremiteit INTEGER NOT NULL,
stijl_toelichting TEXT,
materiele_impact INTEGER NOT NULL,
materiele_toelichting TEXT
)
"""
)
@ -241,14 +241,14 @@ def store_scores(db_path: str, results: list[dict[str, Any]]) -> int:
con.execute(
"""
INSERT OR REPLACE INTO extremity_scores_2d
(motion_id, stylistic_score, material_score, stylistic_rationale, material_rationale)
(motion_id, stijl_extremiteit, stijl_toelichting, materiele_impact, materiele_toelichting)
VALUES (?, ?, ?, ?, ?)
""",
(
r["motion_id"],
r["stijl_extremiteit"],
r["materiele_impact"],
r.get("stijl_toelichting"),
r["materiele_impact"],
r.get("materiele_toelichting"),
),
)

@ -177,7 +177,7 @@ class TestStoreScores:
con = duckdb.connect(tmp_duckdb_path)
try:
rows = con.execute(
"SELECT motion_id, stylistic_score, material_score "
"SELECT motion_id, stijl_extremiteit, materiele_impact "
"FROM extremity_scores_2d ORDER BY motion_id"
).fetchall()
assert len(rows) == 2
@ -206,7 +206,7 @@ class TestStoreScores:
con = duckdb.connect(tmp_duckdb_path)
try:
rows = con.execute(
"SELECT stylistic_score, material_score FROM extremity_scores_2d WHERE motion_id = 1"
"SELECT stijl_extremiteit, materiele_impact FROM extremity_scores_2d WHERE motion_id = 1"
).fetchall()
assert rows[0] == (5, 5)
finally:

Loading…
Cancel
Save