fixes for broken test send script

GrumpyOldTroll · Jun 28, 2021 · d178b6c · d178b6c
1 parent 7840911
commit d178b6c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/send.py b/test/send.py
@@ -5,6 +5,7 @@
 import argparse
 import time
 from ipaddress import ip_address
+from math import ceil
 
 def main(args_in):
     parser = argparse.ArgumentParser(
@@ -69,7 +70,7 @@ def main(args_in):
 
     count = args.count
     if args.duration is not None:
-        count = ceil(duration * 1000 / args.interval)
+        count = ceil(args.duration * 1000 / args.interval)
 
     data = ''.join([f'{i:04x}' for i in range((args.size-1)//4 + 1)])[:args.size].encode()