X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=test.py;fp=test.py;h=e058269e06d49ffb4543c88753d99ec7ff0275e9;hp=ac3eb7e2506e809d640658094587e1df3a61db95;hb=4a824920ad843ba0d1bd3d9b10a8eb0188a2bc63;hpb=7e2780052c458a329fc36253b83eb5194ffb7fed diff --git a/test.py b/test.py index ac3eb7e..e058269 100755 --- a/test.py +++ b/test.py @@ -9,8 +9,8 @@ from _ext import flatland ###################################################################### parser = argparse.ArgumentParser( - description='Dummy test of the flatland sequence generation.', - formatter_class=argparse.ArgumentDefaultsHelpFormatter + description = 'Dummy test of the flatland sequence generation.', + formatter_class = argparse.ArgumentDefaultsHelpFormatter ) parser.add_argument('--seed', @@ -84,7 +84,8 @@ def sequences_to_image(x, gap = 1, gap_color = (0, 128, 255)): ###################################################################### -x = flatland.generate_sequence(args.nb_sequences, +x = flatland.generate_sequence(False, + args.nb_sequences, args.nb_images_per_sequences, args.height, args.width, args.nb_shapes,