Lua 技巧

Lua

Lua 的 if __main__

在 Python 中，使用 if __name__ == '__main__': 可以让某些代码只在直接运行该脚本时执行，而在被当做模块导入时则不执行，很适合单元测试。

在 Lua 中要实现类似功能可以这么做。

if not ({...})[1] then
    -- CODE...
end

CODE 部分的代码只在直接运行该脚本时才被执行，而在被 require 时不会执行。

• 但是用 dofile() 时还是会执行……

单行函数

检查传入参数是否全部相等，支持任意数量的参数。

• 测试一组数是否相等，只要判断这组数当中的最大者和最小者是否相等即可。
• 而 Lua 的 math.min() 和 math.max() 均支持任意数量的参数，所以一句话即可。无需任何遍历。

function all_eq(...) return math.min(...) == math.max(...) end