def METHOD_NAME(self):
    equation = "bmk,bkn->bmn"
    input_dims, output_dim = EinsumDims.parse_equation(equation)
    edims = EinsumDims.parse_dims(input_dims, output_dim)
    self.assertEqual(edims.batch_dims, ["b"])
    self.assertEqual(edims.contracting_dims, ["k"])
    self.assertEqual(edims.lhs_out_only_dims, ["m"])
    self.assertEqual(edims.rhs_out_only_dims, ["n"])
    equation = "bcmk,bckn->bcmn"
    input_dims, output_dim = EinsumDims.parse_equation(equation)
    edims = EinsumDims.parse_dims(input_dims, output_dim)
    self.assertEqual(edims.batch_dims, ["b", "c"])
    self.assertEqual(edims.contracting_dims, ["k"])
    self.assertEqual(edims.lhs_out_only_dims, ["m"])
    self.assertEqual(edims.rhs_out_only_dims, ["n"])