Verifies that L buckets were indeed created
()
| 26 | |
| 27 | |
| 28 | def test_bucketer_cosine(): |
| 29 | """Verifies that L buckets were indeed created""" |
| 30 | L = 7 # number of ORs |
| 31 | bucketer = generate_cosine_lsh_bucketer(d=3, M=5, L=L) |
| 32 | |
| 33 | data_df = pd.DataFrame({"data": [[1, 2, 3], [4, 5, 6]]}) |
| 34 | data = T(data_df, format="pandas", unsafe_trusted_ids=True) |
| 35 | data += data.select(buckets=apply(bucketer, data.data)) |
| 36 | res_pd = table_to_pandas(data) |
| 37 | assert len(res_pd["buckets"].iloc[0]) == L |
| 38 | |
| 39 | |
| 40 | def test_lsh(): |
nothing calls this directly
no test coverage detected