1 files changed, 112 insertions, 0 deletions
diff --git a/Data/Libraries/Penlight/tests/test-func.lua b/Data/Libraries/Penlight/tests/test-func.lua
new file mode 100644
index 0000000..4e43f30
--- /dev/null
+++ b/Data/Libraries/Penlight/tests/test-func.lua
@@ -0,0 +1,112 @@
+local utils = require 'pl.utils'
+local List = require 'pl.List'
+local tablex = require 'pl.tablex'
+asserteq = require('pl.test').asserteq
+utils.import('pl.func')
+
+ -- _DEBUG = true
+
+function pprint (t)
+    print(pretty.write(t))
+end
+
+function test (e)
+    local v = {}
+    print('test',collect_values(e,v))
+    if #v > 0 then pprint(v) end
+    local rep = repr(e)
+    print(rep)
+end
+
+function teste (e,rs,ve)
+    local v = {}
+    collect_values(e,v)
+    if #v > 0 then asserteq(v,ve,nil,1) end
+    local rep = repr(e)
+    asserteq(rep,rs, nil, 1)
+end
+
+teste(_1+_2('hello'),'_1 + _2(_C1)',{"hello"})
+teste(_1:method(),'_1[_C1](_1)',{"method"})
+teste(Not(_1),'not _1')
+
+asserteq(instantiate(_1+_2)(10,20),30)
+asserteq(instantiate(_1+20)(10),30)
+asserteq(instantiate(Or(Not(_1),_2))(true,true),true)
+
+teste(_1() + _2() + _3(),'_1() + _2() + _3()',30)
+asserteq(I(_1+_2)(10,20),30)
+
+teste(_1() - -_2() % _3(), '_1() - - _2() % _3()')
+teste((_1() - -_2()) % _3(), '(_1() - - _2()) % _3()')
+
+teste(_1() - _2() + _3(), '_1() - _2() + _3()')
+teste(_1() - (_2() + _3()), '_1() - (_2() + _3())')
+teste((_1() - _2()) + _3(), '_1() - _2() + _3()')
+
+teste(_1() .. _2() .. _3(), '_1() .. _2() .. _3()')
+teste(_1() .. (_2() .. _3()), '_1() .. _2() .. _3()')
+teste((_1() .. _2()) .. _3(), '(_1() .. _2()) .. _3()')
+
+teste(_1() ^ _2() ^ _3(), '_1() ^ _2() ^ _3()')
+teste(_1() ^ (_2() ^ _3()), '_1() ^ _2() ^ _3()')
+teste((_1() ^ _2()) ^ _3(), '(_1() ^ _2()) ^ _3()')
+
+teste(-_1() * _2(), '- _1() * _2()')
+teste(-(_1() * _2()), '- (_1() * _2())')
+teste((-_1()) * _2(), '- _1() * _2()')
+teste(-_1() ^ _2(), '- _1() ^ _2()')
+teste(-(_1() ^ _2()), '- _1() ^ _2()')
+teste((-_1()) ^ _2(), '(- _1()) ^ _2()')
+
+asserteq(instantiate(_1+_2)(10,20),30)
+
+ls = List {1,2,3,4}
+res = ls:map(10*_1 - 1)
+asserteq(res,List {9,19,29,39})
+
+-- note that relational operators can't be overloaded for _different_ types
+ls = List {10,20,30,40}
+asserteq(ls:filter(Gt(_1,20)),List {30,40})
+
+
+local map,map2 = tablex.map,tablex.map2
+
+--~ test(Len(_1))
+
+-- methods can be applied to all items in a table with map
+asserteq (map(_1:sub(1,2),{'one','four'}),{'on','fo'})
+
+--~ -- or you can do this using List:map
+asserteq( List({'one','four'}):map(_1:sub(1,2)), List{'on','fo'})
+
+--~ -- note that Len can't be represented generally by #, since this can only be overriden by userdata
+asserteq( map(Len(_1),{'one','four'}),  {3,4} )
+
+--~ -- simularly, 'and' and 'or' are not really operators in Lua, so we need a function notation for them
+asserteq (map2(Or(_1,_2),{false,'b'},{'.lua',false}),{'.lua','b'})
+
+--~ -- binary operators:  + - * / % ^ ..
+asserteq (map2(_1.._2,{'a','b'},{'.lua','.c'}),{'a.lua','b.c'})
+
+t1 = {alice=23,fred=34}
+t2 = {bob=25,fred=34}
+
+intersection = bind(tablex.merge,_1,_2,false)
+
+asserteq(intersection(t1,t2),{fred=34})
+
+union = bind(tablex.merge,_1,_2,true)
+
+asserteq(union(t1,t2),{bob=25,fred=34,alice=23})
+
+asserteq(repr(_1+_2),"_1 + _2")
+
+
+
+
+
+
+
+
+