diff options
author | Roman Grundkiewicz <rgrundki@exseed.ed.ac.uk> | 2020-07-26 22:16:33 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-07-26 22:16:33 +0300 |
commit | f5d6369dc51d8d9361e175f48527d566c44b3651 (patch) | |
tree | 2fc8c7d25793c03c0df97949bfc03f5e0b344e65 /tests | |
parent | 7b8f6ee5b6ff7779fd993df7f77adf1e2d9adbe5 (diff) | |
parent | d3885698711406ae5853d3c06fbd47e86f41a8c8 (diff) |
Merge pull request #55 from marian-nmt/input-tsv-server
Add tests for multi-source inputs into marian-server
Diffstat (limited to 'tests')
-rw-r--r-- | tests/interface/input-tsv/test_tsv_decode.sh | 2 | ||||
-rw-r--r-- | tests/interface/input-tsv/test_tsv_server.sh | 32 | ||||
-rw-r--r-- | tests/interface/input-tsv/test_tsv_server_dual_source.sh | 32 |
3 files changed, 65 insertions, 1 deletions
diff --git a/tests/interface/input-tsv/test_tsv_decode.sh b/tests/interface/input-tsv/test_tsv_decode.sh index e3464b0..5deb3ad 100644 --- a/tests/interface/input-tsv/test_tsv_decode.sh +++ b/tests/interface/input-tsv/test_tsv_decode.sh @@ -2,7 +2,7 @@ ##################################################################### # SUMMARY: Translate a single-source input with --tsv option enabled -# TAGS: multi-source transformer sentencepiece tsv +# TAGS: sentencepiece tsv ##################################################################### # Exit on error diff --git a/tests/interface/input-tsv/test_tsv_server.sh b/tests/interface/input-tsv/test_tsv_server.sh new file mode 100644 index 0000000..ad215a5 --- /dev/null +++ b/tests/interface/input-tsv/test_tsv_server.sh @@ -0,0 +1,32 @@ +#!/bin/bash + +##################################################################### +# SUMMARY: Translate a single-source input with --tsv using marian-server +# TAGS: sentencepiece tsv server +##################################################################### + +# Exit on error +set -e + +clean_up() { + kill $SERVER_PID +} +trap clean_up EXIT + +# Remove old artifacts +rm -f server.out + +# Run Marian +$MRT_MARIAN/marian-server -c $MRT_MODELS/rnn-spm/decode.yml -p 8765 -b 6 --tsv > server.log 2>&1 & +SERVER_PID=$! + +sleep 10 + +python3 $MRT_MARIAN/../scripts/server/client_example.py -p 8765 -b 32 < decode.txt > server.out +kill $SERVER_PID + +# Compare outputs +$MRT_TOOLS/diff.sh server.out decode.expected > server.diff + +# Exit with success code +exit 0 diff --git a/tests/interface/input-tsv/test_tsv_server_dual_source.sh b/tests/interface/input-tsv/test_tsv_server_dual_source.sh new file mode 100644 index 0000000..22c253d --- /dev/null +++ b/tests/interface/input-tsv/test_tsv_server_dual_source.sh @@ -0,0 +1,32 @@ +#!/bin/bash + +##################################################################### +# SUMMARY: Translate a single-source input with --tsv using marian-server +# TAGS: multi-source transformer sentencepiece tsv server +##################################################################### + +# Exit on error +set -e + +clean_up() { + kill $SERVER_PID +} +trap clean_up EXIT + +# Remove old artifacts +rm -f server_ape.out + +# Run Marian +$MRT_MARIAN/marian-server -c $MRT_MODELS/ape/config.yml -p 8765 -b 6 --tsv > server_ape.log 2>&1 & +SERVER_PID=$! + +sleep 10 + +python3 $MRT_MARIAN/../scripts/server/client_example.py -p 8765 -b 32 < decode_ape.tsv > server_ape.out +kill $SERVER_PID + +# Compare outputs +$MRT_TOOLS/diff.sh server_ape.out decode_ape.expected > server_ape.diff + +# Exit with success code +exit 0 |