mirror of
https://github.com/nim-lang/Nim.git
synced 2026-04-20 06:20:38 +00:00
added test for symbol type
This commit is contained in:
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)
|
||||
|
||||
Reference in New Issue
Block a user