Function test_bsz_h

tests/test_core.py:53–55

(args)

    ],
)
def test_bsz_h(args):
    h, expected_bsz_h = args
    assert bsz_h(h) == expected_bsz_h


@pytest.mark.parametrize("n_hidden_states", [1, 2])