Added missing unittest execution code (#4400)

same code as on every other test file, just missing from this one for some reason.
This commit is contained in:
George-the-1st 2024-05-02 22:34:30 -04:00 committed by GitHub
parent d4062cb6fc
commit 0627e26140
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions

View File

@ -11,4 +11,7 @@ class TestArange(unittest.TestCase):
f1 = self._get_flops(256)
f2 = self._get_flops(2560)
print(f"{f1=}, {f2=}")
assert f2 / f1 < 15, f"bad complexity, flops {f2/f1:.1f}X while inputs 10X"
assert f2 / f1 < 15, f"bad complexity, flops {f2/f1:.1f}X while inputs 10X"
if __name__ == "__main__":
unittest.main()