Fix missing import, add os.PathLike helptext test

brentyi · Feb 14, 2023 · 0d22057 · 0d22057
1 parent 4abc342
commit 0d22057
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tests/test_helptext.py b/tests/test_helptext.py
@@ -1,5 +1,6 @@
 import dataclasses
 import enum
+import os
 import pathlib
 from collections.abc import Callable
 from typing import Any, Dict, Generic, List, Optional, Tuple, TypeVar, Union, cast
@@ -562,3 +563,11 @@ def main2(x: Callable = nn.ReLU):
     assert "--x {fixed}" in helptext
     assert "(fixed to:" in helptext
     assert "torch" in helptext
+
+
+def test_pathlike() -> None:
+    def main(x: os.PathLike) -> None:
+        pass
+
+    helptext = get_helptext(main)
+    assert "--x PATH " in helptext
diff --git a/tests/test_tyro.py b/tests/test_tyro.py
@@ -2,6 +2,7 @@
 import copy
 import dataclasses
 import enum
+import os
 import pathlib
 from typing import (
     Any,