From 08fc9156ea51458d4f020353b04aff1403132342 Mon Sep 17 00:00:00 2001 From: Sean O'Hair Date: Wed, 16 Oct 2024 12:51:56 -0700 Subject: [PATCH] feat: add smoke test for the age-classification pipeline Signed-off-by: Sean O'Hair --- smoke_test.sh | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/smoke_test.sh b/smoke_test.sh index ea42db7..3693b2d 100755 --- a/smoke_test.sh +++ b/smoke_test.sh @@ -93,7 +93,7 @@ teardown # 3. Loss Prevention GPU results: should see non-empty pipeline0.log contents echo "Running Loss Prevention GPU with logs..." -make run DEVICE_ENV=res/yolov5-gpu.env DEVICE=GPU +make run DEVICE_ENV=res/all-gpu.env DEVICE=GPU status_code=$? verifyStatusCode $status_code # allowing some time to process @@ -109,4 +109,14 @@ verifyStatusCode $status_code # allowing some time to process waitForLogFile verifyNonEmptyPipelineLog +teardown + +# 5. Age pipeline: should see non-empty pipeline0.log contents +echo "Running Age Classification pipeline with logs..." +INPUTSRC=https://www.pexels.com/download/video/3248275 PIPELINE_SCRIPT=age_recognition.sh docker compose -f src/docker-compose.yml up -d +status_code=$? +verifyStatusCode $status_code +# allowing some time to process +waitForLogFile +verifyNonEmptyPipelineLog teardown \ No newline at end of file