add test for db lookup
This commit is contained in:
parent
d0d5f9aa87
commit
29558a734b
1 changed files with 39 additions and 0 deletions
|
|
@ -54,3 +54,42 @@ def test_find_lookup_by_id(tmp_path: Path) -> None:
|
||||||
assert owner == result[3]
|
assert owner == result[3]
|
||||||
assert code == result[4]
|
assert code == result[4]
|
||||||
|
|
||||||
|
|
||||||
|
def test_db_lookup(tmp_path: Path) -> None:
|
||||||
|
db = make_sql(tmp_path)
|
||||||
|
conn = sqlite3.connect(db)
|
||||||
|
c = conn.cursor()
|
||||||
|
|
||||||
|
corr = "f_A"
|
||||||
|
ensemble = "SF_A"
|
||||||
|
code = "openQCD"
|
||||||
|
meas_path = "archive/SF_A/f_A/Project_A.json.gz::asdfasdfasdf"
|
||||||
|
uuid = "Project_A"
|
||||||
|
pars = "{par_A: 3.0, par_B: 5.0}"
|
||||||
|
parameter_file = "projects/Project_A/myinput.in"
|
||||||
|
c.execute("INSERT INTO backlogs (name, ensemble, code, path, project, parameters, parameter_file, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, datetime('now'), datetime('now'))",
|
||||||
|
(corr, ensemble, code, meas_path, uuid, pars, parameter_file))
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
results = find._db_lookup(db, ensemble, corr, code)
|
||||||
|
assert len(results) == 1
|
||||||
|
results = find._db_lookup(db, "SF_B", corr, code)
|
||||||
|
assert results.empty
|
||||||
|
results = find._db_lookup(db, ensemble, "g_A", code)
|
||||||
|
assert results.empty
|
||||||
|
results = find._db_lookup(db, ensemble, corr, "sfcf")
|
||||||
|
assert results.empty
|
||||||
|
results = find._db_lookup(db, ensemble, corr, code, project = "Project_A")
|
||||||
|
assert len(results) == 1
|
||||||
|
results = find._db_lookup(db, ensemble, corr, code, project = "Project_B")
|
||||||
|
assert results.empty
|
||||||
|
results = find._db_lookup(db, ensemble, corr, code, parameters = pars)
|
||||||
|
assert len(results) == 1
|
||||||
|
results = find._db_lookup(db, ensemble, corr, code, parameters = "{par_A: 3.0, par_B: 4.0}")
|
||||||
|
assert results.empty
|
||||||
|
#results = find._db_lookup(db, ensemble, corr, code, project, parameters, created_before, created_after, updated_before, updated_after)
|
||||||
|
|
||||||
|
#results = find._db_lookup(db, ensemble, corr, code, project, parameters, created_before, created_after, updated_before, updated_after)
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue