From 42e215baae5ba124cf4aa66a79bf68d4fe2e29c1 Mon Sep 17 00:00:00 2001 From: Rob Austein Date: Sat, 13 Apr 2024 20:58:03 -0400 Subject: Rewrite to handle corner case start == finish + 1 --- test1.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) (limited to 'test1.py') diff --git a/test1.py b/test1.py index 737d550..e374bb7 100755 --- a/test1.py +++ b/test1.py @@ -1,9 +1,14 @@ #!/usr/bin/env python3 from rfc1982_serial_number import Serial, find_intermediate +from argparse import ArgumentParser from random import randint -for test in range(10000): +ap = ArgumentParser() +ap.add_argument("iterations", type = int, default = 100000, nargs = "?") +args = ap.parse_args() + +for test in range(args.iterations): i1 = Serial(randint(0, Serial.modulus - 1)) i2 = Serial(randint(0, Serial.modulus - 1)) @@ -13,8 +18,11 @@ for test in range(10000): int(i1) ^ int(i2) == Serial.modulus >> 1 step = [i1, i2] - wrap = find_intermediate(i1, i2) - if wrap is not None: - step.insert(1, wrap) + for wrap in find_intermediate(i1, i2): + assert wrap > step[-2] + step.insert(-1, wrap) + + for i in range(len(step) - 1): + assert step[i] < step[i + 1] print(" => ".join(str(s) for s in step)) -- cgit v1.2.3