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

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()