test_cases = [
    {
        "xs": [[1.0, 1.0], [0.0, 1.0]],
        "ys": [[1.0, 1.0], [0.0, 1.0]],
        "result": {'align_loss': 0.0, 'x_unif_loss': -2.0, 'y_unif_loss': -2.0, 'unif_loss': -2.0}
    }
        
]