(self)
| 126 | self.do_forward(network) |
| 127 | |
| 128 | def test_network_basic(self): |
| 129 | network = LiteNetwork() |
| 130 | network.load(self.model_path) |
| 131 | |
| 132 | input_name = network.get_input_name(0) |
| 133 | input_tensor = network.get_io_tensor(input_name) |
| 134 | output_name = network.get_output_name(0) |
| 135 | output_tensor = network.get_io_tensor(output_name) |
| 136 | |
| 137 | assert input_tensor.layout.shapes[0] == 1 |
| 138 | assert input_tensor.layout.shapes[1] == 3 |
| 139 | assert input_tensor.layout.shapes[2] == 224 |
| 140 | assert input_tensor.layout.shapes[3] == 224 |
| 141 | assert input_tensor.layout.data_type == LiteDataType.LITE_FLOAT |
| 142 | assert input_tensor.layout.ndim == 4 |
| 143 | |
| 144 | self.do_forward(network) |
| 145 | |
| 146 | def test_network_shared_data(self): |
| 147 | network = LiteNetwork() |
nothing calls this directly
no test coverage detected