improve table.wrap

1 files changed, 16 insertions, 0 deletions

diff --git a/tests/modules/table/test.lua b/tests/modules/table/test.lua
index 5ed17620c..426dfe516 100644
--- a/tests/modules/table/test.lua
+++ b/tests/modules/table/test.lua
@@ -9,3 +9,19 @@ function test_find_if(t)
     t:are_equal(table.find({1, 2, 4, 4, 5, 6}, 4), {3, 4})
     t:are_equal(table.find_first({1, 2, 3, 4, 5, 6}, 4), 4)
 end
+
+function test_wrap()
+    t:are_equal(talbe.wrap(1), {1})
+    t:are_equal(talbe.wrap(nil), {})
+    t:are_equal(talbe.wrap({}), {})
+    t:are_equal(talbe.wrap({1}), {1})
+    t:are_equal(talbe.wrap({{}}), {{}})
+end
+
+function test_unwrap()
+    t:are_equal(talbe.unwrap(1), 1)
+    t:are_equal(talbe.unwrap(nil), nil)
+    t:are_equal(talbe.unwrap({}), nil)
+    t:are_equal(talbe.unwrap({1}), 1)
+    t:are_equal(talbe.unwrap({{}}), {})
+end