MNT print narx to markdown table (#133)

Author: MatthewSZhang

.github/workflows/test.yml

@@ -49,3 +49,4 @@ jobs:
         uses: codecov/codecov-action@v5
         with:
           token: ${{ secrets.CODECOV_TOKEN }}
+          verbose: true

fastcan/narx/_base.py

@@ -132,7 +132,7 @@ class NARX(MultiOutputMixin, RegressorMixin, BaseEstimator):
     ...             delay_ids=delay_ids).fit(X, y, coef_init="one_step_ahead")
     >>> print_narx(narx)
     | yid |        Term        |   Coef   |
-    =======================================
+    |-----|--------------------|----------|
     |  0  |     Intercept      |  1.008   |
     |  0  |    y_hat[k-1,0]    |  0.498   |
     |  0  |      X[k-2,0]      |  0.701   |

fastcan/narx/_utils.py

@@ -69,7 +69,7 @@ def print_narx(
     >>> X, y = load_diabetes(return_X_y=True)
     >>> print_narx(NARX().fit(X, y), term_space=10, coef_space=5, float_precision=0)
     | yid |   Term   |Coef |
-    ========================
+    |-----|----------|-----|
     |  0  |Intercept | 152 |
     |  0  |  X[k,0]  | -10 |
     |  0  |  X[k,1]  |-240 |
@@ -102,7 +102,7 @@ def _get_term_str(term_feat_ids, term_delay_ids):
         + f"|{'Term':^{term_space}}"
         + f"|{'Coef':^{coef_space}}|"
     )
-    print("=" * (yid_space + term_space + coef_space + 4))
+    print(f"|{'-'*yid_space}|{'-'*term_space}|{'-'*coef_space}|")
     for i in range(narx.n_outputs_):
         print(
             f"|{i:^{yid_space}}|"
@@ -246,7 +246,7 @@ def make_narx(
     0.0289
     >>> print_narx(narx)
     | yid |        Term        |   Coef   |
-    =======================================
+    |-----|--------------------|----------|
     |  0  |     Intercept      |  1.050   |
     |  0  |    y_hat[k-1,0]    |  0.484   |
     |  0  |   X[k,0]*X[k,0]    |  0.306   |

fastcan/narx/tests/test_narx.py

@@ -517,6 +517,8 @@ def test_print_narx(capsys):
     captured = capsys.readouterr()
     # Check if the header is present in the output
     assert "| yid |        Term        |   Coef   |" in captured.out
+    # Check if the separator line is present
+    assert "|-----|--------------------|----------|" in captured.out
     # Check if the intercept line for yid 0 is present
     assert "|  0  |     Intercept      |" in captured.out
     # Check if the intercept line for yid 1 is present