def test_label_field_can_index_with_vocab(self):
vocab = Vocabulary()
vocab.add_token_to_namespace("entailment", namespace="labels")
vocab.add_token_to_namespace("contradiction", namespace="labels")
vocab.add_token_to_namespace("neutral", namespace="labels")
label = LabelField("entailment")
label.index(vocab)
array = label.as_array(label.get_padding_lengths())
numpy.testing.assert_array_almost_equal(array, numpy.array([0]))
def test_as_array_returns_integer_array(self):
label = LabelField(5, skip_indexing=True)
array = label.as_array(label.get_padding_lengths())
numpy.testing.assert_array_almost_equal(array, numpy.array([5]))
def test_pad_returns_one_hot_array(self):
label = LabelField(5, num_labels=10)
array = label.as_array(label.get_padding_lengths())
numpy.testing.assert_array_almost_equal(
array, numpy.array([0, 0, 0, 0, 0, 1, 0, 0, 0, 0]))
