Spaces:
Running
Running
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)
|