expected_output = {
        aname:
            np.array(  # pylint: disable=too-many-function-args
                [[1, 1], [-1, -1]],
                dtype=np.float32).reshape(2, 1, 2, 1),