From 737cbd28f1107aa0922bcb620c72dd04dd1549f2 Mon Sep 17 00:00:00 2001 From: "Chanwut (Mick) Kittivorawong" Date: Sun, 13 Aug 2023 15:18:43 -0700 Subject: [PATCH] save test --- .github/workflows/test.yml | 3 +++ tests/video_processor/steps/test_detection_3d.py | 8 ++++---- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 9903428..3bbd5e4 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -166,6 +166,9 @@ jobs: with: name: video-processor-coverage path: ./coverage.xml + + - name: Commit formatted changes + run: ./scripts/commit-and-push.sh "[CI] update test" test-engine-and-interface: name: Test Engine and Interface diff --git a/tests/video_processor/steps/test_detection_3d.py b/tests/video_processor/steps/test_detection_3d.py index 11a2dd3..9633232 100644 --- a/tests/video_processor/steps/test_detection_3d.py +++ b/tests/video_processor/steps/test_detection_3d.py @@ -47,10 +47,10 @@ def test_detection_3d(): det_result = FromDetection2DAndRoad.get(output) assert det_result is not None - # with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.json'), 'w') as f: - # json.dump([(d[0].cpu().numpy().tolist(), d[1], d[2]) for d in det_result], f, indent=1) - # with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.pkl'), 'wb') as f: - # pickle.dump([(d[0].cpu(), d[1], d[2]) for d in det_result], f) + with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.json'), 'w') as f: + json.dump([(d[0].cpu().numpy().tolist(), d[1], d[2]) for d in det_result], f, indent=1) + with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.pkl'), 'wb') as f: + pickle.dump([(d[0].cpu(), d[1], d[2]) for d in det_result], f) with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.pkl'), 'rb') as f: det_groundtruth = pickle.load(f)