55 lines
2 KiB
Python
55 lines
2 KiB
Python
import corrlib.integrity as integ
|
|
import corrlib.find as find
|
|
import datalad.api as dl
|
|
import corrlib.initialization as cinit
|
|
import sqlite3
|
|
from pathlib import Path
|
|
import os
|
|
|
|
|
|
def test_list_ensembles(tmp_path: Path) -> None:
|
|
"""
|
|
Check against the implementation in find to check if they are the same.
|
|
"""
|
|
os.mkdir(tmp_path / 'archive')
|
|
os.mkdir(tmp_path / 'archive' / 'A')
|
|
os.mkdir(tmp_path / 'archive' / 'B')
|
|
os.mkdir(tmp_path / 'archive' / 'C')
|
|
integ_results = integ._list_ensembles(tmp_path)
|
|
assert len(integ_results) == 3
|
|
find_results = find.list_ensembles(tmp_path)
|
|
assert len(find_results) == 3
|
|
for f,i in zip(find_results, integ_results):
|
|
assert f == i
|
|
|
|
|
|
def test_list_projects(tmp_path: Path) -> None:
|
|
cinit.create(tmp_path)
|
|
db = tmp_path / "backlogger.db"
|
|
dl.unlock(str(db), dataset=str(tmp_path))
|
|
conn = sqlite3.connect(db)
|
|
c = conn.cursor()
|
|
|
|
customTags = ""
|
|
owner = "owner"
|
|
code = "sfcf"
|
|
created_at = "today"
|
|
updated_at = "today"
|
|
|
|
id = "asdf1"
|
|
aliases = "a1,s1,d1,f1"
|
|
c.execute("INSERT INTO projects (id, aliases, customTags, owner, code, created_at, updated_at) VALUES (?,?,?,?,?,?,?)", (id, aliases, customTags, owner, code , created_at, updated_at))
|
|
id = "asdf2"
|
|
aliases = "a2,s2,d2,f2"
|
|
c.execute("INSERT INTO projects (id, aliases, customTags, owner, code, created_at, updated_at) VALUES (?,?,?,?,?,?,?)", (id, aliases, customTags, owner, code , created_at, updated_at))
|
|
id = "asdf3"
|
|
aliases = "a3,s3,d3,f3"
|
|
c.execute("INSERT INTO projects (id, aliases, customTags, owner, code, created_at, updated_at) VALUES (?,?,?,?,?,?,?)", (id, aliases, customTags, owner, code , created_at, updated_at))
|
|
conn.commit()
|
|
conn.close
|
|
integ_results = integ._list_projects(tmp_path)
|
|
assert len(integ_results) == 3
|
|
find_results = find.list_projects(tmp_path)
|
|
assert len(find_results) == 3
|
|
for f,i in zip(find_results, integ_results):
|
|
assert f == i
|