From 5d50e19432788483c15411be18de107fbfdf5763 Mon Sep 17 00:00:00 2001
From: Jeong YunWon <jeong@youknowone.org>
Date: Wed, 8 Jan 2020 11:52:49 +0900
Subject: [PATCH] dict comprehension test

---
 parser/src/parser.rs | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/parser/src/parser.rs b/parser/src/parser.rs
index 48db5846a..7a2ddd248 100644
--- a/parser/src/parser.rs
+++ b/parser/src/parser.rs
@@ -393,6 +393,31 @@ mod tests {
         )
     }
 
+    #[test]
+    fn test_parse_dict_comprehension() {
+        let source = String::from("{x1: x2 for y in z}");
+        let parse_ast = parse_expression(&source).unwrap();
+        assert_eq!(
+            parse_ast,
+            ast::Expression {
+                location: ast::Location::new(1, 1),
+                node: ast::ExpressionType::Comprehension {
+                    kind: Box::new(ast::ComprehensionKind::Dict {
+                        key: mk_ident("x1", 1, 2),
+                        value: mk_ident("x2", 1, 6),
+                    }),
+                    generators: vec![ast::Comprehension {
+                        location: ast::Location::new(1, 9),
+                        target: mk_ident("y", 1, 13),
+                        iter: mk_ident("z", 1, 18),
+                        ifs: vec![],
+                        is_async: false,
+                    }],
+                }
+            }
+        );
+    }
+
     #[test]
     fn test_parse_list_comprehension() {
         let source = String::from("[x for y in z]");