test: update assert

XuehaiPan · XuehaiPan · commit a31ff81dc160 · 2022-08-09T15:25:06.000+08:00
diff --git a/Makefile b/Makefile
@@ -83,7 +83,7 @@ addlicense-install: go-install
 
 pytest: pytest-install
 	cd tests && \
-	$(PYTHON) -m pytest --verbose --color=yes --maxfail=16 \
+	$(PYTHON) -m pytest --verbose --color=yes --maxfail=32 \
 		--numprocesses=logical --maxprocesses=8 --durations=0 \
 		--cov="../$(PROJECT_PATH)" --cov-report=term-missing \
 		.
diff --git a/tests/helpers.py b/tests/helpers.py
@@ -94,6 +94,6 @@ def assert_all_close(
 ) -> None:
 
     assert torch.allclose(input, other, rtol=rtol, atol=atol, equal_nan=equal_nan), (
-        f'L_inf = {(input - other).abs().max()}, '
+        f'L_inf = {(input - other).abs().max():.5g}, '
         f'fail_rate = {torch.logical_not((input - other).abs() <= atol + rtol * other.abs()).float().mean()}'
     )

Original file line number	Diff line number	Diff line change
`@@ -83,7 +83,7 @@ addlicense-install: go-install`
`83`	`83`
`84`	`84`	`pytest: pytest-install`
`85`	`85`	`cd tests && \`
`86`		`- $(PYTHON) -m pytest --verbose --color=yes --maxfail=16 \`
	`86`	`+ $(PYTHON) -m pytest --verbose --color=yes --maxfail=32 \`
`87`	`87`	`--numprocesses=logical --maxprocesses=8 --durations=0 \`
`88`	`88`	`--cov="../$(PROJECT_PATH)" --cov-report=term-missing \`
`89`	`89`	`.`
Original file line number	Diff line number	Diff line change
`@@ -94,6 +94,6 @@ def assert_all_close(`
`94`	`94`	`) -> None:`
`95`	`95`
`96`	`96`	`assert torch.allclose(input, other, rtol=rtol, atol=atol, equal_nan=equal_nan), (`
`97`		`- f'L_inf = {(input - other).abs().max()}, '`
	`97`	`+ f'L_inf = {(input - other).abs().max():.5g}, '`
`98`	`98`	`f'fail_rate = {torch.logical_not((input - other).abs() <= atol + rtol * other.abs()).float().mean()}'`
`99`	`99`	`)`