extern crate python_parser;
use python_parser::ast::*;
use python_parser::*;
#[test]
fn assign_listcomp() {
let (rest, ast) = file_input(make_strspan("foo = [bar for baz in qux]")).unwrap();
assert_eq!(rest.fragment.0, "");
assert_eq!(
ast,
vec![Statement::Assignment(
vec![Expression::Name("foo".to_string()),],
vec![vec![Expression::ListComp(
Box::new(SetItem::Unique(Expression::Name("bar".to_string()))),
vec![ComprehensionChunk::For {
async: false,
item: vec![Expression::Name("baz".to_string())],
iterator: Expression::Name("qux".to_string()),
}]
)]]
)]
);
}