Skip to content

Commit 03491ca

Browse files
tests/gentle_compare.py: make pixmaps_rms() reuse rms().
1 parent 0f08ebc commit 03491ca

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

tests/gentle_compare.py

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -28,13 +28,17 @@ def gentle_compare(w0, w1):
2828
return True
2929

3030

31-
def rms(a, b):
31+
def rms(a, b, verbose=None, out_prefix=''):
3232
'''
3333
Returns RMS diff of raw bytes of two sequences.
3434
'''
35+
if verbose is True:
36+
verbose = 100000
3537
assert len(a) == len(b)
3638
e = 0
37-
for aa, bb in zip(a, b):
39+
for i, (aa, bb) in enumerate(zip(a, b)):
40+
if verbose and (i % verbose == 0):
41+
print(f'{out_prefix}rms(): {i=} {e=} {aa=} {aa=}.')
3842
e += (aa - bb) ** 2
3943
rms = math.sqrt(e / len(a))
4044
return rms
@@ -58,14 +62,9 @@ def pixmaps_rms(a, b, out_prefix=''):
5862
a_mv = a.samples_mv
5963
b_mv = b.samples_mv
6064
assert len(a_mv) == len(b_mv)
61-
e = 0
62-
for i, (a_byte, b_byte) in enumerate(zip(a_mv, b_mv)):
63-
if i % 100000 == 0:
64-
print(f'{out_prefix}compare_pixmaps(): {i=} {e=} {a_byte=} {b_byte=}.')
65-
e += (a_byte - b_byte) ** 2
66-
rms = math.sqrt(e / len(a_mv))
67-
print(f'{out_prefix}compare_pixmaps(): {e=} {rms=}.')
68-
return rms
65+
ret = rms(a_mv, b_mv, verbose=True, out_prefix=out_prefix)
66+
print(f'{out_prefix}pixmaps_rms(): {ret=}.')
67+
return ret
6968

7069

7170
def pixmaps_diff(a, b, out_prefix=''):

0 commit comments

Comments
 (0)