Function test_bitlinear_shapes

tests/tests.py:60–62

(in_features, out_features)

    "in_features,out_features", [(10, 20), (20, 40), (5, 10), (15, 10)]
)
def test_bitlinear_shapes(in_features, out_features):
    layer = BitLinear(in_features, out_features)
    assert layer.weight.shape == (out_features, in_features)


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