mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change metadata_test to tolerate paths with backslashes.

Browse Source 
Fixes a test failure under Windows.

The backslashes end up doubled in the MLIR string because of escaping.
This commit is contained in:
Peter Hawkins 2023-06-12 21:57:41 -04:00
parent bfc7a6ddf3
commit c787b3da07
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/metadata_test.py
View File

@ -94,17 +94,17 @@ class MetadataTest(jtu.JaxTestCase):
)
# Sanity check
self.assertIn("/tests/metadata_test.py", make_hlo())
self.assertRegex(make_hlo(), r"[/\\]+tests[/\\]+metadata_test.py")
with jax_config.hlo_source_file_canonicalization_regex(".*/tests/"):
with jax_config.hlo_source_file_canonicalization_regex(r".*[\\/]+tests[/\\]+"):
hlo = make_hlo()
self.assertIn("metadata_test.py", hlo)
self.assertNotIn("tests/", hlo)
self.assertNotIn("/metadata_test.py", hlo)
self.assertNotRegex(hlo, r"tests[/\\]+")
self.assertNotRegex(hlo, r"[/\\]+metadata_test.py")
with jax_config.hlo_source_file_canonicalization_regex("no_match_xxx"):
hlo = make_hlo()
self.assertIn("/tests/metadata_test.py", hlo)
self.assertRegex(hlo, r"[/\\]+tests[/\\]+metadata_test.py")
with jax_config.hlo_source_file_canonicalization_regex(".*"):
hlo = make_hlo()
@ -112,7 +112,8 @@ class MetadataTest(jtu.JaxTestCase):
with jax_config.hlo_source_file_canonicalization_regex("test"):
hlo = make_hlo()
self.assertIn("/s/metadata_.py", hlo)
self.assertRegex(hlo, r"[/\\]+s[/\\]+metadata_.py")
if __name__ == "__main__":
absltest.main(testLoader=jtu.JaxTestLoader())