Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit f6cd5a9

Browse files
committed
fix event analysis
1 parent bc79255 commit f6cd5a9

File tree

1 file changed

+5
-14
lines changed

1 file changed

+5
-14
lines changed

src/eegprep/eeg_compare.py

Lines changed: 5 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -41,15 +41,6 @@ def isequaln(a, b):
4141
return bool(np.all(b == a))
4242
except:
4343
pass
44-
# Handle nested structures
45-
if isinstance(a, dict) and isinstance(b, dict):
46-
if set(a.keys()) != set(b.keys()):
47-
return False
48-
return all(isequaln(a[k], b[k]) for k in a.keys())
49-
elif isinstance(a, list) and isinstance(b, list):
50-
if len(a) != len(b):
51-
return False
52-
return all(isequaln(a[i], b[i]) for i in range(len(a)))
5344
# Final comparison - ensure we return a boolean
5445
try:
5546
result = a == b
@@ -60,7 +51,7 @@ def isequaln(a, b):
6051
return False
6152

6253
"""Compare two EEG-like structures, reporting differences to stderr."""
63-
print('Field analysis:')
54+
print('\nField analysis: (no entries means OK)')
6455
# Handle both dictionary-like objects and objects with __dict__
6556
if hasattr(eeg1, 'keys'):
6657
# Dictionary-like object
@@ -87,13 +78,13 @@ def isequaln(a, b):
8778
print(f' Field {field} differs (ok, supposed to differ)')
8879
elif any(sub in name for sub in ('subject', 'session', 'run', 'task')):
8980
print(f' Field {field} differs ("{v1}" vs "{v2}")', file=sys.stderr)
90-
elif any(sub in name for sub in ('eventdescription', 'event')):
81+
elif any(sub in name for sub in ('chanlocs', 'event', 'reject')):
82+
pass
83+
# For complex nested structures, provide more detailed info
84+
elif any(sub in name for sub in ('eventdescription')):
9185
n1 = len(v1) if isinstance(v1, Sequence) else 1
9286
n2 = len(v2) if isinstance(v2, Sequence) else 1
9387
print(f' Field {field} differs (n={n1} vs n={n2})', file=sys.stderr)
94-
elif any(sub in name for sub in ('chanlocs', 'reject')):
95-
# For complex nested structures, provide more detailed info
96-
print(f' Field {field} differs (complex structure - see detailed analysis below)')
9788
else:
9889
print(f' Field {field} differs', file=sys.stderr)
9990
# compare xmin/xmax

0 commit comments

Comments
 (0)