Nim/tests/macros/ttypenodes.nim
2021-11-24 16:34:42 +01:00

16 lines
430 B
Nim

import macros
macro makeEnum(): untyped =
newTree(nnkEnumTy, newEmptyNode(), ident"a", ident"b", ident"c")
macro makeObject(): untyped =
newTree(nnkObjectTy, newEmptyNode(), newEmptyNode(), newTree(nnkRecList,
newTree(nnkIdentDefs, ident"x", ident"y", ident"int", newEmptyNode())))
type
Foo = makeEnum()
Bar = makeObject()
doAssert {a, b, c} is set[Foo]
let bar = Bar(x: 3, y: 4)
doAssert (bar.x, bar.y) == (3, 4)