Spaces:
Sleeping
Sleeping
MilesCranmer
commited on
Commit
•
c7c02bf
1
Parent(s):
11f524f
Soft fail tests which require determinism
Browse files- test/test.py +22 -2
test/test.py
CHANGED
@@ -350,7 +350,17 @@ class TestMiscellaneous(unittest.TestCase):
|
|
350 |
model = PySRRegressor(
|
351 |
max_evals=10000, verbosity=0, progress=False
|
352 |
) # Return early.
|
|
|
|
|
|
|
353 |
check_generator = check_estimator(model, generate_only=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
354 |
exception_messages = []
|
355 |
for (_, check) in check_generator:
|
356 |
try:
|
@@ -359,10 +369,20 @@ class TestMiscellaneous(unittest.TestCase):
|
|
359 |
check(model)
|
360 |
print("Passed", check.func.__name__)
|
361 |
except Exception as e:
|
362 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
363 |
print("Failed", check.func.__name__, "with:")
|
364 |
# Add a leading tab to error message, which
|
365 |
# might be multi-line:
|
366 |
-
print("\n".join([(" " * 4) + row for row in
|
367 |
# If any checks failed don't let the test pass.
|
368 |
self.assertEqual([], exception_messages)
|
|
|
350 |
model = PySRRegressor(
|
351 |
max_evals=10000, verbosity=0, progress=False
|
352 |
) # Return early.
|
353 |
+
|
354 |
+
# TODO: Add deterministic option so that we can test these.
|
355 |
+
# (would require backend changes, and procs=0 for serialism.)
|
356 |
check_generator = check_estimator(model, generate_only=True)
|
357 |
+
tests_requiring_determinism = [
|
358 |
+
"check_regressors_int", # PySR is not deterministic, so fails this.
|
359 |
+
"check_regressor_data_not_an_array",
|
360 |
+
"check_supervised_y_2d",
|
361 |
+
"check_regressors_int",
|
362 |
+
"check_fit_idempotent",
|
363 |
+
]
|
364 |
exception_messages = []
|
365 |
for (_, check) in check_generator:
|
366 |
try:
|
|
|
369 |
check(model)
|
370 |
print("Passed", check.func.__name__)
|
371 |
except Exception as e:
|
372 |
+
error_message = str(e)
|
373 |
+
failed_tolerance_check = "Not equal to tolerance" in error_message
|
374 |
+
|
375 |
+
if (
|
376 |
+
failed_tolerance_check
|
377 |
+
and check.func.__name__ in tests_requiring_determinism
|
378 |
+
):
|
379 |
+
# Skip test as PySR is not deterministic.
|
380 |
+
continue
|
381 |
+
|
382 |
+
exception_messages.append(f"{check.func.__name__}: {error_message}\n")
|
383 |
print("Failed", check.func.__name__, "with:")
|
384 |
# Add a leading tab to error message, which
|
385 |
# might be multi-line:
|
386 |
+
print("\n".join([(" " * 4) + row for row in error_message.split("\n")]))
|
387 |
# If any checks failed don't let the test pass.
|
388 |
self.assertEqual([], exception_messages)
|