Show method

JuliaDynamics · Aug 3, 2024 · 7a2fb29 · 7a2fb29
1 parent ea3f73c
commit 7a2fb29
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/methods/correlation/azadkia_chatterjee_coefficient.jl.jl b/src/methods/correlation/azadkia_chatterjee_coefficient.jl.jl
@@ -74,6 +74,12 @@ struct AzadkiaChatterjeeCoefficient{V, L, R, MI, NI, RM, RN} <: AssociationMeasu
     theiler::Int
 end
 
+function Base.show(io::IO, m::AzadkiaChatterjeeCoefficient)
+    theiler = m.theiler
+    print(io, "AzadkiaChatterjeeCoefficient(; theiler = $theiler)")
+end
+
+
 min_inputs_vars(::AzadkiaChatterjeeCoefficient) = 2
 max_inputs_vars(::AzadkiaChatterjeeCoefficient) = 3
 

diff --git a/test/methods/correlation/azadkia_chatterjee_coefficient.jl b/test/methods/correlation/azadkia_chatterjee_coefficient.jl
@@ -7,4 +7,8 @@ y = rand(rng, n)
 z = rand(rng, n)
 m = AzadkiaChatterjeeCoefficient()
 @test association(m, x, y) isa Real
-@test association(m, x, y, z) isa Real
+@test association(m, x, y, z) isa Real
+# pretty printing
+out = repr(AzadkiaChatterjeeCoefficient())
+@test occursin("AzadkiaChatterjeeCoefficient", out)
+@test occursin("theiler = ", out)