Adapt example to API changes.

1 files changed, 16 insertions, 6 deletions

diff --git a/example.py b/example.py
index 52b99f3..ecdef1d 100644
--- a/example.py
+++ b/example.py
@@ -18,16 +18,26 @@
 import bgproc
 
 
-N = 10
+import sys
+N = int(sys.argv[1])
 
 
-def inc(request):
-    return request + 1
+def func(request):
+    return request
 
 
-bg = bgproc.BackgroundProcessing(inc)
-for i in range(N):
+bg = bgproc.BackgroundProcessing(func, numprocs=2)
+numbers = range(N)
+results = []
+for i in numbers:
     bg.enqueue_request(i)
+    results.extend(bg.get_results())
 bg.close_requests()
-print bg.get_results(True)
+results.extend(bg.get_results(block_all=True))
+if numbers != sorted(results):
+    print 'N:', N
+    print 'len(numbers):', len(numbers)
+    print 'len(results):', len(results)
+    raise Exception('did not get all results')
+bg.finish()