def test2():
data = np.array([[1.0, 1.0, 1.0, 0.8, 0.8, 1., 0., 0.],
[0.5, 0.5, 0.5, 0.2, 0.2, 0., 1., 1.],
[1.5, 1.5, 1.5, 1.2, 1.2, 1., 0., 1.]], dtype=np.float32)
ws = np.array([1.0, 0.8, 0.3], dtype=np.float32)
ds = np.array([3, 2, 3])
fs = ['cosine_s', 'cosine_s', 'jaccard_s']
sims = nn.query(data[0,:], data, ws, ds, fs)
np.testing.assert_almost_equal(sims, calc_sims(data[0,:], data, ws, ds, fs))
def test1(size):
data = np.random.rand(size, 6).astype(dtype=np.float32)
ws = np.array([1.0, 0.3], dtype=np.float32)
ds = np.array([3, 3])
sims = nn.query(data[0,:], data, ws, ds)
