move assert gpu_times not nan

slefrancois · slefrancois · commit f724c2c6054c · 2016-10-07T17:38:37.000-04:00
diff --git a/code/test.py b/code/test.py
@@ -238,3 +238,6 @@ def write_junit(f, algos, times, label):
         for label, times in times_dic.items():
             write_junit(f, algo_executed, times, label)
         f.write('</testsuite>\n')
+
+    if do_gpu:
+        assert not numpy.isnan(gpu_times).any()
