feat: empty constructor for DifferentialEquation and HarmonicEquation with Jac

Author: oameye

.github/workflows/Downstream.yml

@@ -29,8 +29,6 @@ jobs:
   test:
     name: ${{ matrix.package.repo }}/${{ matrix.julia-version }}
     runs-on: ${{ matrix.os }}
-    env:
-      GROUP: ${{ matrix.package.group }}
     strategy:
       fail-fast: false
       matrix:

src/DifferentialEquation.jl

@@ -32,6 +32,10 @@ mutable struct DifferentialEquation
         return new(eqs, OrderedDict(var => OrderedSet() for var in keys(eqs)))
     end
 
+    function DifferentialEquation()
+        return new(OrderedDict{Num,Equation}(), OrderedDict{Num,OrderedSet{Num}}())
+    end
+
     # uses the above constructor if no harmonics defined
     function DifferentialEquation(eqs::Vector{Equation}, vars::Vector{Num})
         if length(eqs) != length(vars)

src/HarmonicEquation.jl

@@ -19,20 +19,29 @@ mutable struct HarmonicEquation
     jacobian::Matrix{Num}
 
     # use a self-referential constructor with _parameters
-    function HarmonicEquation(equations, variables, nat_eq)
+    function HarmonicEquation(
+        equations::Vector{Equation},
+        variables::Vector{HarmonicVariable},
+        nat_eq::DifferentialEquation,
+    )
         return (
-            x=new(
+            x = new(
                 equations,
                 variables,
                 Num[],
                 nat_eq,
                 dummy_symbolic_Jacobian(length(variables)),
             );
-            x.parameters=_parameters(x);
+            x.parameters = _parameters(x);
             x
         )
     end
-    function HarmonicEquation(equations, variables, parameters, natural_equation)
+    function HarmonicEquation(
+        equations::Vector{Equation},
+        variables::Vector{HarmonicVariable},
+        parameters::Vector{Num},
+        natural_equation::DifferentialEquation,
+    )
         return new(
             equations,
             variables,
@@ -41,6 +50,20 @@ mutable struct HarmonicEquation
             dummy_symbolic_Jacobian(length(variables)),
         )
     end
+    function HarmonicEquation(
+        equations::Vector{Equation},
+        variables::Vector{HarmonicVariable},
+        parameters::Vector{Num},
+        jacobian::Matrix{Num},
+    )
+        return new(
+            equations,
+            variables,
+            parameters,
+            DifferentialEquation(),
+            jacobian,
+        )
+    end
 end
 
 "Get the parameters (not time nor variables) of a HarmonicEquation"

test/DifferentialEquations.jl

@@ -42,6 +42,11 @@ using QuestBase:
     diff_eq3 = DifferentialEquation(expr, x)
     @test length(diff_eq3.equations) == 1
     @test diff_eq3.equations[x].rhs == 0
+
+    # Test empty constructor
+    diff = DifferentialEquation()
+    @test isempty(diff.equations)
+    @test isempty(diff.harmonics)
 end
 
 @testset "Helper Functions" begin

test/HarmonicEquation.jl

@@ -46,6 +46,9 @@ hv2 = HarmonicVariable(v, "test", "v", Num(1.0), y)
         @test heq.parameters == Num[]
         @test heq.jacobian isa Matrix{Num}
     end
+
+    heq3 = HarmonicEquation([eq1, eq2], [hv1, hv2], Num[], Num[1 1; 1 1])
+    @test isempty(heq3.natural_equation.harmonics)
 end
 
 @testset "Parameter handling" begin