Started to implement the gravity by making the pulling optional.

[flatland.git] / test.py

diff --git a/test.py b/test.py
index ac3eb7e..e058269 100755 (executable)
--- 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,