gpython/ast/dump_test.go at main · go-python/gpython · GitHub

39 lines (35 loc) · 1.4 KB

// Use of this source code is governed by a BSD-style

// license that can be found in the LICENSE file.

package ast

import (

"testing"

"github.com/go-python/gpython/py"

)

func TestDump(t *testing.T) {

for _, test := range []struct {

in Ast

out string

}{

{nil, `<nil>`},

{&Pass{}, `Pass()`},

{&Str{S: py.String("potato")}, `Str(s='potato')`},

{&Bytes{S: py.Bytes("potato")}, `Bytes(s=b'potato')`},

{&BinOp{Left: &Str{S: py.String("one")}, Op: Add, Right: &Str{S: py.String("two")}},

`BinOp(left=Str(s='one'), op=Add(), right=Str(s='two'))`},

{&Module{}, `Module(body=[])`},

{&Module{Body: []Stmt{&Pass{}}}, `Module(body=[Pass()])`},

{&Module{Body: []Stmt{&ExprStmt{Value: &Tuple{}}}}, `Module(body=[Expr(value=Tuple(elts=[], ctx=UnknownExprContext(0)))])`},

{&NameConstant{Value: py.True}, `NameConstant(value=True)`},

{&Name{Id: Identifier("hello"), Ctx: Load}, `Name(id='hello', ctx=Load())`},

{&ListComp{Elt: &Str{S: py.String("potato")}, Generators: []Comprehension{{

Target: &Name{Id: Identifier("hello"), Ctx: Load},

}}}, `ListComp(elt=Str(s='potato'), generators=[comprehension(target=Name(id='hello', ctx=Load()), iter=None, ifs=[])])`},

} {

out := Dump(test.in)

if out != test.out {

t.Errorf("Dump(%#v) got %q expected %q", test.in, out, test.out)

}

Expand file tree