Update strategy_test.py