TEST: Modify tests for coverage

QuantEcon · Apr 9, 2020 · 8515484 · 8515484
1 parent 6461b5d
commit 8515484
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/quantecon/tests/test_amf.py b/quantecon/tests/test_amf.py
@@ -144,7 +144,7 @@ def test_invalid_dimensions(self):
     def test_invalid_shape(self):
         inputs = (self.A, self.B, self.D, self.F, self.ν)
         inputs_passed = list(inputs)
-        invalid_input = np.eye(10)
+        invalid_input = np.ones((10, 10, 10))
 
         for i in range(len(inputs)):
             inputs_passed[i] = invalid_input  # Set input i to be invalid
@@ -163,6 +163,12 @@ def test_non_square_A(self):
         with assert_raises(ValueError):
             AMF_LSS_VAR(A, B, D, F, ν)
 
+    def test_default_kwargs(self):
+        amf = AMF_LSS_VAR(self.A, self.B, self.D)
+
+        assert_array_equal(amf.F, np.zeros((amf.nk, amf.nk)))
+        assert_array_equal(amf.ν, np.zeros((amf.ny, 1)))
+
     def test_loglikelihood(self):
         x = np.random.rand(4, 10) * 0.005
         y = np.random.rand(1, 10) * 0.005