diff --git a/tests/pydyad_spsc/consumer.py b/tests/pydyad_spsc/consumer.py index 19cd5f42..d13ca1b4 100644 --- a/tests/pydyad_spsc/consumer.py +++ b/tests/pydyad_spsc/consumer.py @@ -30,7 +30,7 @@ def consume_data(i, num_ints_expected): ) if not np.array_equal(int_buf, verify_int_buf): raise RuntimeError("Consumed data is incorrect!") - print("Correctly consumed data (iter {})".format(i)) + print("Correctly consumed data (iter {})".format(i), flush=True) def main(): @@ -45,7 +45,7 @@ def main(): global CONS_DIR CONS_DIR = args.cons_managed_dir.expanduser().resolve() for i in range(args.num_files): - print("Trying to consume data (iter {})".format(i)) + print("Trying to consume data (iter {})".format(i), flush=True) consume_data(i, args.num_ints_expected) diff --git a/tests/pydyad_spsc/producer.py b/tests/pydyad_spsc/producer.py index 63b77a2c..f5f13daa 100644 --- a/tests/pydyad_spsc/producer.py +++ b/tests/pydyad_spsc/producer.py @@ -17,7 +17,7 @@ def produce_data(i, num_ints): int_buf = np.arange(start_val, start_val+num_ints, dtype=np.uint64) with dyad_open(fname, "wb") as f: np.save(f, int_buf) - print("Successfully produced data (iter {})".format(i)) + print("Successfully produced data (iter {})".format(i), flush=True) def main(): @@ -32,7 +32,7 @@ def main(): global PROD_DIR PROD_DIR = args.prod_managed_dir.expanduser().resolve() for i in range(args.num_files): - print("Trying to produce data (iter {})".format(i)) + print("Trying to produce data (iter {})".format(i), flush=True) produce_data(i, args.num_ints)