def test_subset(rng, algebra): v1 = Vocabulary(32, pointer_gen=rng, algebra=algebra) v1.populate('A; B; C; D; E; F; G') # Test creating a vocabulary subset v2 = v1.create_subset(['A', 'C', 'E']) assert list(v2.keys()) == ['A', 'C', 'E'] assert_equal(v2['A'].v, v1['A'].v) assert_equal(v2['C'].v, v1['C'].v) assert_equal(v2['E'].v, v1['E'].v) assert v1.algebra is v2.algebra
def test_subset(rng, algebra): v1 = Vocabulary(32, pointer_gen=rng, algebra=algebra) v1.populate("A; B; C; D; E; F; G") # Test creating a vocabulary subset v2 = v1.create_subset(["A", "C", "E"]) assert list(v2.keys()) == ["A", "C", "E"] assert_equal(v2["A"].v, v1["A"].v) assert_equal(v2["C"].v, v1["C"].v) assert_equal(v2["E"].v, v1["E"].v) assert v1.algebra is v2.algebra