def test_sample_with_seed(): da = DocumentArray(random_docs(100)) sampled_1 = da.sample(5, seed=1) sampled_2 = da.sample(5, seed=1) sampled_3 = da.sample(5, seed=2) assert len(sampled_1) == len(sampled_2) == len(sampled_3) == 5 assert sampled_1 == sampled_2 assert sampled_1 != sampled_3
def test_sample(): da = DocumentArray(random_docs(100)) sampled = da.sample(1) assert len(sampled) == 1 sampled = da.sample(5) assert len(sampled) == 5 assert isinstance(sampled, DocumentArray) with pytest.raises(ValueError): da.sample(101) # can not sample with k greater than lenth of document array.