Added expand_vocab_gradient test.