Prolog: only catch timeout in test functions

7 files changed, 7 insertions, 7 deletions

diff --git a/prolog/problems/sets/diff_3/common.py b/prolog/problems/sets/diff_3/common.py
index d9b5362..fe21e50 100644
--- a/prolog/problems/sets/diff_3/common.py
+++ b/prolog/problems/sets/diff_3/common.py
@@ -48,7 +48,7 @@ def test(code, aux_code):
             for query, answers in test_cases:
                 if prolog.engine.check_answers(engine_id, query=query, answers=answers, timeout=1.0):
                     n_correct += 1
-    except:
+    except socket.timeout:
         pass
     finally:
         if engine_id:
diff --git a/prolog/problems/sets/intersect_3/common.py b/prolog/problems/sets/intersect_3/common.py
index e30e3dc..1cf2945 100644
--- a/prolog/problems/sets/intersect_3/common.py
+++ b/prolog/problems/sets/intersect_3/common.py
@@ -47,7 +47,7 @@ def test(code, aux_code):
             for query, answers in test_cases:
                 if prolog.engine.check_answers(engine_id, query=query, answers=answers, timeout=1.0):
                     n_correct += 1
-    except:
+    except socket.timeout:
         pass
     finally:
         if engine_id:
diff --git a/prolog/problems/sets/is_subset_2/common.py b/prolog/problems/sets/is_subset_2/common.py
index aee4e81..696a4a6 100644
--- a/prolog/problems/sets/is_subset_2/common.py
+++ b/prolog/problems/sets/is_subset_2/common.py
@@ -41,7 +41,7 @@ def test(code, aux_code):
             for query, answers in test_cases:
                 if prolog.engine.check_answers(engine_id, query=query, answers=answers, timeout=1.0):
                     n_correct += 1
-    except:
+    except socket.timeout:
         pass
     finally:
         if engine_id:
diff --git a/prolog/problems/sets/is_superset_2/common.py b/prolog/problems/sets/is_superset_2/common.py
index db32c8d..41a0aa3 100644
--- a/prolog/problems/sets/is_superset_2/common.py
+++ b/prolog/problems/sets/is_superset_2/common.py
@@ -41,7 +41,7 @@ def test(code, aux_code):
             for query, answers in test_cases:
                 if prolog.engine.check_answers(engine_id, query=query, answers=answers, timeout=1.0):
                     n_correct += 1
-    except:
+    except socket.timeout:
         pass
     finally:
         if engine_id:
diff --git a/prolog/problems/sets/powerset_2/common.py b/prolog/problems/sets/powerset_2/common.py
index 020b05e..aec5cab 100644
--- a/prolog/problems/sets/powerset_2/common.py
+++ b/prolog/problems/sets/powerset_2/common.py
@@ -39,7 +39,7 @@ def test(code, aux_code):
             for query, answers in test_cases:
                 if prolog.engine.check_answers(engine_id, query=query, answers=answers, timeout=1.0):
                     n_correct += 1
-    except:
+    except socket.timeout:
         pass
     finally:
         if engine_id:
diff --git a/prolog/problems/sets/subset_2/common.py b/prolog/problems/sets/subset_2/common.py
index 3871880..9cf4714 100644
--- a/prolog/problems/sets/subset_2/common.py
+++ b/prolog/problems/sets/subset_2/common.py
@@ -52,7 +52,7 @@ def test(code, aux_code):
             for query, answers in test_cases:
                 if prolog.engine.check_answers(engine_id, query=query, answers=answers, timeout=1.0):
                     n_correct += 1
-    except:
+    except socket.timeout:
         pass
     finally:
         if engine_id:
diff --git a/prolog/problems/sets/union_3/common.py b/prolog/problems/sets/union_3/common.py
index 30780dc..899d20e 100644
--- a/prolog/problems/sets/union_3/common.py
+++ b/prolog/problems/sets/union_3/common.py
@@ -44,7 +44,7 @@ def test(code, aux_code):
             for query, answers in test_cases:
                 if prolog.engine.check_answers(engine_id, query=query, answers=answers, timeout=1.0):
                     n_correct += 1
-    except:
+    except socket.timeout:
         pass
     finally:
         if engine_id: