def test_call_shapes():
 assert_equal(lae_a(y).shape, (n_y,))
 assert_equal(lae_b(y).shape, (n_y,))

def test_call_shapes ( ) : assert_equal ( lae_a ( y ) . shape , ( n_y , ) ) assert_equal ( lae_b ( y ) . shape , ( n_y , ) )
lae: shape of call to lae .