Skip to content

Commit def5752

Browse files
tests: Added test_as_database_list
1 parent f1cd5c5 commit def5752

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-1
lines changed

opxrd/database/opxrd.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,9 @@ def as_database_list(cls, root_dirpath : str, download : bool = True) -> list[Pa
2929

3030
pattern_dbs = []
3131
for d in os.listdir(path=root_dirpath):
32-
db = PatternDB.load(dirpath=os.path.join(root_dirpath, d), strict=True)
32+
dirpath = os.path.join(root_dirpath, d)
33+
db = PatternDB.load(dirpath=dirpath, strict=True)
34+
db.name = d
3335
pattern_dbs.append(db)
3436
return pattern_dbs
3537

tests/t_download.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,20 @@ def test_dl(self):
2727
self.check_pattern_data(fpath,p)
2828
k.increment()
2929

30+
def test_as_database_list(self):
31+
dl_dirpath = tempfile.mktemp()
32+
dbs = OpXRD.as_database_list(root_dirpath=dl_dirpath, download=True)
33+
print(f'- Checking database loading')
34+
self.assertTrue(os.path.isdir(dl_dirpath))
35+
self.assertTrue(len(dbs) > 5)
36+
37+
print(f'- Checking pattern data ok')
38+
k = TrackedInt(start_value=0,finish_value=len(dbs))
39+
for db in dbs:
40+
for p in db.patterns:
41+
self.check_pattern_data(fpath='None',p=p)
42+
k.increment()
43+
3044
@staticmethod
3145
def check_pattern_data(fpath: str, p: XrdPattern):
3246
try:

0 commit comments

Comments
 (0)