Add KF into test cases

acse-jm122 · Jan 12, 2025 · a7a57f2 · a7a57f2
1 parent 7a98ea7
commit a7a57f2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_deepda.py b/tests/test_deepda.py
@@ -50,6 +50,7 @@ def executor(torchda):
 
 
 def test_algorithms(algorithms):
+    assert algorithms.KF in algorithms
     assert algorithms.EnKF in algorithms
     assert algorithms.Var3D in algorithms
     assert algorithms.Var4D in algorithms
@@ -114,9 +115,11 @@ def test_case_set_parameter(case):
 
 
 def test_case_set_algorithm(case, algorithms):
+    assert case.set_algorithm(algorithms.KF)
     assert case.set_algorithm(algorithms.EnKF)
     assert case.set_algorithm(algorithms.Var3D)
     assert case.set_algorithm(algorithms.Var4D)
+    assert case.set_parameter("algorithm", algorithms.KF)
     assert case.set_parameter("algorithm", algorithms.EnKF)
     assert case.set_parameter("algorithm", algorithms.Var3D)
     assert case.set_parameter("algorithm", algorithms.Var4D)