def testMultipleInputs(self):
inputs_info = {
"x": tensor_info.ParsedTensorInfo(
tf.float32,
tf.TensorShape([None]),
is_sparse=False),
"y": tensor_info.ParsedTensorInfo(
tf.float32,
tf.TensorShape([None]),
is_sparse=False),
}
def _check(dict_inputs):
self.assertEqual(len(dict_inputs), 2)
for key in ("x", "y"):
self.assertEqual(dict_inputs[key].dtype, tf.float32)
self.assertTrue(dict_inputs[key].shape.is_compatible_with([None]))
_check(module._convert_dict_inputs({"x": [1, 2], "y": [1, 2]},
inputs_info))
with self.assertRaisesRegexp(TypeError, r"missing \['x', 'y'\]"):
module._convert_dict_inputs(None, inputs_info)
with self.assertRaisesRegexp(TypeError, r"missing \['x', 'y'\]"):
module._convert_dict_inputs({}, inputs_info)
with self.assertRaisesRegexp(TypeError, r"missing \['x', 'y'\]"):
module._convert_dict_inputs({"z": 1}, inputs_info)
with self.assertRaisesRegexp(
TypeError, "Signature expects multiple inputs. Use a dict."):
module._convert_dict_inputs(1, inputs_info)
def testNoInputs(self):
self.assertEqual(module._convert_dict_inputs(None, {}), {})
self.assertEqual(module._convert_dict_inputs({}, {}), {})
with self.assertRaisesRegexp(TypeError, "expects no inputs"):
module._convert_dict_inputs([None], {})
with self.assertRaisesRegexp(TypeError, "expects no inputs"):
module._convert_dict_inputs(1, {})
with self.assertRaisesRegexp(TypeError, r"extra given \['x'\]"):
module._convert_dict_inputs({"x": 1}, {})
def testSingleInput(self):
inputs_info = {
"x": tensor_info.ParsedTensorInfo(
tf.float32,
tf.TensorShape([None]),
is_sparse=False),
}
def _check(dict_inputs):
self.assertEqual(len(dict_inputs), 1)
self.assertEqual(dict_inputs["x"].dtype, tf.float32)
self.assertTrue(dict_inputs["x"].shape.is_compatible_with([None]))
_check(module._convert_dict_inputs([1, 2], inputs_info))
_check(module._convert_dict_inputs({"x": [1, 2]}, inputs_info))
with self.assertRaisesRegexp(TypeError, r"missing \['x'\]"):
module._convert_dict_inputs(None, inputs_info)
with self.assertRaisesRegexp(TypeError, r"extra given \['y'\]"):
module._convert_dict_inputs({"x": [1, 2], "y": [1, 2]}, inputs_info)
