luatz/timetable: Optimize and export `is_leap`; add tests

[etc/awesome.git] / luatz / timetable.lua

diff --git a/luatz/timetable.lua b/luatz/timetable.lua
index 10af17018f90a9f14f5ef0c2ce72af9f9e367616..99ac52ee8d238f85e350e5e4ed2a92b1fa6b32b6 100644 (file)
--- a/luatz/timetable.lua
+++ b/luatz/timetable.lua
@@ -16,7 +16,13 @@ end
 local sakamoto = {0, 3, 2, 5, 0, 3, 5, 1, 4, 6, 2, 4};
 
 local function is_leap ( y )
-       return (y % 4) == 0 and (y % 100) ~= 0 or (y % 400) == 0
+       if (y % 4) ~= 0 then
+               return false
+       elseif (y % 100) ~= 0 then
+               return true
+       else
+               return (y % 400) == 0
+       end
 end
 
 local function year_length ( y )
@@ -193,10 +199,14 @@ function timetable_methods:clone ( )
 end
 
 local function new_from_timestamp ( ts )
+       if type ( ts ) ~= "number" then
+               error ( "bad argument #1 to 'new_from_timestamp' (number expected, got " .. type ( ts ) .. ")" , 2 )
+       end
        return new_timetable ( 1970 , 1 , 1 , 0 , 0 , ts )
 end
 
 return {
+       is_leap = is_leap ;
        day_of_year = day_of_year ;
        day_of_week = day_of_week ;
        normalise = normalise ;