fix getattr pretty

Author: saulshanabrook

python/egglog/pretty.py

@@ -473,10 +473,8 @@ def _call_inner(  # noqa: C901, PLR0911, PLR0912
                         return f"{slf}[{self(args[0], unwrap_lit=True)}] = {self(args[1], unwrap_lit=True)}"
                     case _ if method_name in NAMED_UNARY_METHODS:
                         return NAMED_UNARY_METHODS[method_name], [non_str_slf, *args]
-                    case "__getattr__":
-                        if isinstance(args[0], LitDecl) and isinstance(args[0].value, str):
-                            return f"{slf}.{args[0].value}"
-                        return "getattr", [non_str_slf, *args]
+                    case "__getattr__" if isinstance(args[0], LitDecl) and isinstance(args[0].value, str):
+                        return f"{slf}.{args[0].value}"
                     case _:
                         return f"{slf}.{method_name}", args
             case ConstantRef(Ident(name)):

python/tests/test_pretty.py

@@ -169,7 +169,8 @@ def __repr__(self) -> str:
     pytest.param(A() >= A(), "A() >= A()", id="greater than or equal"),
     pytest.param(A() == A(), "A() == A()", id="equal"),
     pytest.param(A() != A(), "A() != A()", id="not equal"),
-    pytest.param(A().foo, 'getattr(A(), "foo")', id="getattr"),
+    pytest.param(A().foo, "A().foo", id="getattr"),
+    pytest.param(A().__getattr__(String("a") + "b"), 'A().__getattr__(join("a", "b"))', id="getattr expr"),
     pytest.param(+A(), "+A()", id="pos"),
     pytest.param(~A(), "~A()", id="invert"),
     pytest.param(round(A()), "round(A())", id="round"),