added test for symbol type
This commit is contained in:
parent
e80840c40a
commit
cfbb7390e2
1 changed files with 49 additions and 0 deletions
49
tests/macros/typesapi2.nim
Normal file
49
tests/macros/typesapi2.nim
Normal file
|
|
@ -0,0 +1,49 @@
|
||||||
|
# tests to see if a symbol returned from macros.getType() can
|
||||||
|
# be used as a type
|
||||||
|
import macros
|
||||||
|
|
||||||
|
macro testTypesym (t:stmt): expr =
|
||||||
|
var ty = t.getType
|
||||||
|
if ty.typekind == ntyTypedesc:
|
||||||
|
# skip typedesc get to the real type
|
||||||
|
ty = ty[1].getType
|
||||||
|
|
||||||
|
if ty.kind == nnkSym: return ty
|
||||||
|
assert ty.kind == nnkBracketExpr
|
||||||
|
assert ty[0].kind == nnkSym
|
||||||
|
result = ty[0]
|
||||||
|
return
|
||||||
|
|
||||||
|
type TestFN = proc(a,b:int):int
|
||||||
|
var iii: testTypesym(TestFN)
|
||||||
|
static: assert iii is TestFN
|
||||||
|
|
||||||
|
proc foo11 : testTypesym(void) =
|
||||||
|
echo "HI!"
|
||||||
|
static: assert foo11 is proc():void
|
||||||
|
|
||||||
|
var sss: testTypesym(seq[int])
|
||||||
|
static: assert sss is seq[int]
|
||||||
|
# very nice :>
|
||||||
|
|
||||||
|
static: assert array[2,int] is testTypesym(array[2,int])
|
||||||
|
static: assert(ref int is testTypesym(ref int))
|
||||||
|
static: assert(void is testTypesym(void))
|
||||||
|
|
||||||
|
|
||||||
|
macro tts2 (t:stmt, idx:int): expr =
|
||||||
|
var ty = t.getType
|
||||||
|
if ty.typekind == ntyTypedesc:
|
||||||
|
# skip typedesc get to the real type
|
||||||
|
ty = ty[1].getType
|
||||||
|
|
||||||
|
if ty.kind == nnkSym: return ty
|
||||||
|
assert ty.kind == nnkBracketExpr
|
||||||
|
return ty[idx.intval.int]
|
||||||
|
type TestFN2 = proc(a:int,b:float):string
|
||||||
|
static:
|
||||||
|
assert(tts2(TestFN2, 0) is TestFN2)
|
||||||
|
assert(tts2(TestFN2, 1) is string)
|
||||||
|
assert(tts2(TestFN2, 2) is int)
|
||||||
|
assert(tts2(TestFN2, 3) is float)
|
||||||
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue