run_test_example.sh 702 字节
Newer Older
J
jingqinghe 已提交
1 2
#!/bin/bash

J
update  
jingqinghe 已提交
3
# set redis server ip and port for test
H
bug_fix  
heya02 已提交
4 5
export TEST_REDIS_IP=${LOCALHOST}
export TEST_REDIS_PORT=${REDIS_PORT}
J
jingqinghe 已提交
6

J
update  
jingqinghe 已提交
7
# unittest command
J
jingqinghe 已提交
8 9
PYTHON_TEST="python -m unittest"

J
update  
jingqinghe 已提交
10 11 12 13 14 15 16 17 18 19 20 21 22
# add the modules to test
TEST_MODULES=("test_datautils_aby3"
"test_datautils_align"
"test_op_add"
"test_op_sub"
"test_op_mul"
"test_op_square"
"test_op_sum"
"test_op_mean"
"test_op_square_error_cost"
"test_op_fc"
"test_op_relu"
"test_op_compare"
J
jhjiangcs 已提交
23 24 25 26 27
"test_input_embedding"
"test_op_softmax_with_cross_entropy"
"test_op_batch_norm"
"test_op_conv"
"test_op_pool"
H
He, Kai 已提交
28
"test_op_metric"
H
He, Kai 已提交
29
"test_data_preprocessing"
30 31
"test_op_reshape"
"test_op_reduce_sum"
J
update  
jingqinghe 已提交
32
)
J
jingqinghe 已提交
33

J
update  
jingqinghe 已提交
34 35
# run unittest
for MODULE in ${TEST_MODULES[@]}
J
jingqinghe 已提交
36 37 38
do
    $PYTHON_TEST $MODULE
done