(self, spec_args, op_args)
| 110 | self.assertEqual(spec_blob, op_blob) |
| 111 | |
| 112 | def assertArgsEqual(self, spec_args, op_args): |
| 113 | self.assertEqual(len(spec_args), len(op_args)) |
| 114 | keys = [a.name for a in op_args] |
| 115 | |
| 116 | def parse_args(args): |
| 117 | operator = caffe2_pb2.OperatorDef() |
| 118 | # Generate the expected value in the same order |
| 119 | for k in keys: |
| 120 | v = args[k] |
| 121 | arg = utils.MakeArgument(k, v) |
| 122 | operator.arg.add().CopyFrom(arg) |
| 123 | return operator.arg |
| 124 | |
| 125 | self.assertEqual(parse_args(spec_args), op_args) |
| 126 | |
| 127 | def assertNetContainOps(self, net, op_specs): |
| 128 | """ |
no test coverage detected