def METHOD_NAME(self):
    """Test with mock inputs to verify that output is as expected"""
    module = OneLayerModule()
    module_inputs = (torch.tensor([[1.0, 2.0, 3.0, 4.0, 5.0, 6.0]]),)
    loc_out = module(*module_inputs)
    expected_output = torch.tensor(
        [
            [
                -3.382883310317993,
                -1.907626986503601,
                -0.662156879901886,
                0.5104053020477295,
                0.9551836252212524,
                1.6657130718231201,
            ]
        ]
    )
    tolerance = 1e-04
    assert tolerance > torch.norm(loc_out - expected_output)