Fix support for MOI.TimeLimitSec (#140)

ds4dm · Sep 5, 2023 · 71aeb53 · 71aeb53
1 parent 069cdda
commit 71aeb53
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/src/Interfaces/MOI/attributes.jl b/src/Interfaces/MOI/attributes.jl
@@ -58,10 +58,13 @@ end
 #
 #   TimeLimitSec
 #
-MOI.get(m::Optimizer, ::MOI.TimeLimitSec) = m.inner.params.IPM.TimeLimit
+function MOI.get(m::Optimizer, ::MOI.TimeLimitSec)
+    value = m.inner.params.IPM.TimeLimit
+    return value == Inf ? nothing : value
+end
 
-function MOI.set(m::Optimizer, ::MOI.TimeLimitSec, t)
-    m.inner.params.IPM.TimeLimit = t
+function MOI.set(m::Optimizer, ::MOI.TimeLimitSec, t::Union{Real,Nothing})
+    m.inner.params.IPM.TimeLimit = convert(Float64, something(t, Inf))
     return nothing
 end
 

diff --git a/test/Interfaces/MOI_wrapper.jl b/test/Interfaces/MOI_wrapper.jl
@@ -110,3 +110,15 @@ MOIU.@model(ModelData,
             "test_unbounded",
         ])
 end
+
+@testset "test_attribute_TimeLimitSec" begin
+    model = Tulip.Optimizer()
+    @test MOI.supports(model, MOI.TimeLimitSec())
+    @test MOI.get(model, MOI.TimeLimitSec()) === nothing
+    MOI.set(model, MOI.TimeLimitSec(), 0.0)
+    @test MOI.get(model, MOI.TimeLimitSec()) == 0.0
+    MOI.set(model, MOI.TimeLimitSec(), nothing)
+    @test MOI.get(model, MOI.TimeLimitSec()) === nothing
+    MOI.set(model, MOI.TimeLimitSec(), 1.0)
+    @test MOI.get(model, MOI.TimeLimitSec()) == 1.0
+end