Update test_skip_gram.py