(self)
| 94 | self.assertTrue((expected_output == output_data).all()) |
| 95 | |
| 96 | def testString(self): |
| 97 | interpreter = interpreter_wrapper.Interpreter( |
| 98 | model_path=resource_loader.get_path_to_datafile( |
| 99 | 'testdata/gather_string.tflite')) |
| 100 | interpreter.allocate_tensors() |
| 101 | |
| 102 | input_details = interpreter.get_input_details() |
| 103 | self.assertEqual(2, len(input_details)) |
| 104 | self.assertEqual('input', input_details[0]['name']) |
| 105 | self.assertEqual(np.string_, input_details[0]['dtype']) |
| 106 | self.assertTrue(([10] == input_details[0]['shape']).all()) |
| 107 | self.assertEqual((0.0, 0), input_details[0]['quantization']) |
| 108 | self.assertEqual('indices', input_details[1]['name']) |
| 109 | self.assertEqual(np.int64, input_details[1]['dtype']) |
| 110 | self.assertTrue(([3] == input_details[1]['shape']).all()) |
| 111 | self.assertEqual((0.0, 0), input_details[1]['quantization']) |
| 112 | |
| 113 | output_details = interpreter.get_output_details() |
| 114 | self.assertEqual(1, len(output_details)) |
| 115 | self.assertEqual('output', output_details[0]['name']) |
| 116 | self.assertEqual(np.string_, output_details[0]['dtype']) |
| 117 | self.assertTrue(([3] == output_details[0]['shape']).all()) |
| 118 | self.assertEqual((0.0, 0), output_details[0]['quantization']) |
| 119 | |
| 120 | test_input = np.array([1, 2, 3], dtype=np.int64) |
| 121 | interpreter.set_tensor(input_details[1]['index'], test_input) |
| 122 | |
| 123 | test_input = np.array(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']) |
| 124 | expected_output = np.array([b'b', b'c', b'd']) |
| 125 | interpreter.set_tensor(input_details[0]['index'], test_input) |
| 126 | interpreter.invoke() |
| 127 | |
| 128 | output_data = interpreter.get_tensor(output_details[0]['index']) |
| 129 | self.assertTrue((expected_output == output_data).all()) |
| 130 | |
| 131 | |
| 132 | class InterpreterTestErrorPropagation(test_util.TensorFlowTestCase): |
nothing calls this directly
no test coverage detected