Skip to content

Commit 8ade1a8

Browse files
style(pre-commit): autofix
1 parent 880519c commit 8ade1a8

File tree

1 file changed

+12
-7
lines changed

1 file changed

+12
-7
lines changed

common/tensorrt_common/test/test_tensorrt_common.cpp

+12-7
Original file line numberDiff line numberDiff line change
@@ -12,18 +12,22 @@
1212
// See the License for the specific language governing permissions and
1313
// limitations under the License.
1414

15-
#include <gtest/gtest.h>
1615
#include "tensorrt_common/tensorrt_common.hpp"
1716

17+
#include <gtest/gtest.h>
18+
1819
// test get_input_dims function
19-
TEST(TrtCommonTest, TestGetInputDims) {
20-
std::string onnx_file_path = "src/universe/autoware.universe/common/tensorrt_common/data/yolov5s.onnx";
20+
TEST(TrtCommonTest, TestGetInputDims)
21+
{
22+
std::string onnx_file_path =
23+
"src/universe/autoware.universe/common/tensorrt_common/data/yolov5s.onnx";
2124
nvinfer1::Dims input_dims = tensorrt_common::get_input_dims(onnx_file_path);
2225
ASSERT_GT(input_dims.nbDims, 0);
2326
}
2427

2528
// test is_valid_precision_string function
26-
TEST(TrtCommonTest, TestIsValidPrecisionString) {
29+
TEST(TrtCommonTest, TestIsValidPrecisionString)
30+
{
2731
std::string valid_precision = "fp16";
2832
std::string invalid_precision = "invalid_precision";
2933
ASSERT_TRUE(tensorrt_common::is_valid_precision_string(valid_precision));
@@ -32,7 +36,8 @@ TEST(TrtCommonTest, TestIsValidPrecisionString) {
3236

3337
// In the future, more test cases will be written to test the functionality of TrtCommon class
3438

35-
int main(int argc, char * argv[]) {
36-
testing::InitGoogleTest(&argc, argv);
37-
return RUN_ALL_TESTS();
39+
int main(int argc, char * argv[])
40+
{
41+
testing::InitGoogleTest(&argc, argv);
42+
return RUN_ALL_TESTS();
3843
}

0 commit comments

Comments
 (0)