From 66c02b47114322f609b104a7dcb15fb7cadf1f11 Mon Sep 17 00:00:00 2001
From: Oscar Dowson <odow@users.noreply.github.com>
Date: Tue, 5 Nov 2024 14:09:26 +1300
Subject: [PATCH] Improve code coverage of src/print.jl (#3867)

---
 test/test_print.jl | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/test/test_print.jl b/test/test_print.jl
index 8e758a61097..20d2c3ba358 100644
--- a/test/test_print.jl
+++ b/test/test_print.jl
@@ -17,6 +17,8 @@ using JuMP
 using LinearAlgebra
 using Test
 
+include("JuMPExtension.jl")
+
 # Helper function to test IO methods work correctly
 function _io_test_show(::MIME"text/plain", obj, exp_str)
     @test sprint(show, obj) == exp_str
@@ -1136,4 +1138,41 @@ function test_small_number_latex()
     return
 end
 
+function test_print_summary()
+    model = JuMPExtension.MyModel()
+    @variable(model, x)
+    @objective(model, Min, x)
+    @test occursin("Minimization problem with:", sprint(show, model))
+    return
+end
+
+function test_show_objective_summary()
+    model = Model()
+    @variable(model, x)
+    @objective(model, Min, x)
+    @test sprint(show_objective_function_summary, model) ==
+          "Objective function type: JuMP.VariableRef\n"
+    @NLobjective(model, Min, x)
+    @test sprint(show_objective_function_summary, model) ==
+          "Objective function type: Nonlinear\n"
+    return
+end
+
+function test_show_constraints_summary()
+    model = Model()
+    @variable(model, x)
+    @constraint(model, x >= 1)
+    @NLconstraint(model, sin(x) == 1)
+    @test sprint(show_constraints_summary, model) ==
+          "`JuMP.AffExpr`-in-`MathOptInterface.GreaterThan{Float64}`: 1 constraint\nNonlinear: 1 constraint\n"
+    return
+end
+
+function test_show_backend_summary()
+    model = Model()
+    @test sprint(show_backend_summary, model) ==
+          "Model mode: AUTOMATIC\nCachingOptimizer state: NO_OPTIMIZER\nSolver name: No optimizer attached."
+    return
+end
+
 end  # TestPrint