def test_assign_arrays_raise_error(): foo = Embedding("foo", [0.1, 0.3, 0.10]) bar = Embedding("bar", [0.7, 0.2, 0.11]) emb = EmbeddingSet(foo, bar) with pytest.raises(ValueError): emb_with_property = emb.assign(prop_a=["a", "b"], prop_b=np.array([1, 2, 3]))
def test_assign_literal_values(): foo = Embedding("foo", [0.1, 0.3, 0.10]) bar = Embedding("bar", [0.7, 0.2, 0.11]) emb = EmbeddingSet(foo, bar) emb_with_property = emb.assign(prop_a="prop-one", prop_b=1) assert all([e.prop_a == "prop-one" for e in emb_with_property]) assert all([e.prop_b == 1 for e in emb_with_property])
def test_assign(): foo = Embedding("foo", [0.1, 0.3, 0.10]) bar = Embedding("bar", [0.7, 0.2, 0.11]) emb = EmbeddingSet(foo, bar) emb_with_property = emb.assign(prop_a=lambda d: "prop-one", prop_b=lambda d: "prop-two") assert all([e.prop_a == "prop-one" for e in emb_with_property]) assert all([e.prop_b == "prop-two" for e in emb_with_property])
def test_assign_arrays(): foo = Embedding("foo", [0.1, 0.3, 0.10]) bar = Embedding("bar", [0.7, 0.2, 0.11]) emb = EmbeddingSet(foo, bar) emb_with_property = emb.assign(prop_a=["a", "b"], prop_b=np.array([1, 2])) assert emb_with_property["foo"].prop_a == "a" assert emb_with_property["bar"].prop_a == "b" assert emb_with_property["foo"].prop_b == 1 assert emb_with_property["bar"].prop_b == 2