fix: ifs without else at the end of comprehensions were not unparsed correctly (#6)

Author: jvllmr

src/unparser.rs

@@ -1152,7 +1152,16 @@ impl Unparser {
         self.write_str(" in ");
         self.unparse_expr(&node.iter);
         for if_ in &node.ifs {
-            self.unparse_expr(if_);
+            self.write_str(" ");
+            match if_ {
+                Expr::Name(name) => {
+                    self.write_str("if ");
+                    self.write_str(&name.id);
+                }
+                _ => {
+                    self.unparse_expr(if_);
+                }
+            }
         }
     }
 

test_files/simple_comprehensions.py

@@ -5,3 +5,9 @@
 set_ = {def_ for def_ in random.choices(string.ascii_letters)}
 dict_ = {k: v for k, v in enumerate(random.choices(string.ascii_letters))}
 list_ = [value for value in random.choices(string.ascii_letters)]
+
+
+with_outer_if = (value for value in random.choices(string.ascii_letters) if value)
+with_inner_if_else = (
+    value if value else "missing!" for value in random.choices(string.ascii_letters)
+)