import { expect } from 'chai';

import { JaccardSimilarity } from '../../code-search/similar/algorithm/JaccardSimilarity';

describe('JaccardSimilarity', () => {
	let jaccardSimilarity: JaccardSimilarity;

	beforeEach(() => {
		jaccardSimilarity = new JaccardSimilarity();
	});

	describe('tokenLevelJaccardSimilarity', () => {
		it('should return correct similarity scores', () => {
			const query = 'test query';
			const chunks = [
				['test', 'query'],
				['another', 'test'],
				['query', 'test'],
			];
			const result = jaccardSimilarity.computeInputSimilarity(query, chunks);
			expect(result).to.be.an('array');
			expect(result[0][0]).to.be.closeTo(0.5, 0.5);
			expect(result[1][0]).to.be.closeTo(0, 0.05);
			expect(result[2][0]).to.be.closeTo(0.5, 0.5);
		});
	});

	describe('tokenize', () => {
		it('should return a set of tokens', () => {
			const input = 'test input';
			const result = jaccardSimilarity['tokenize'](input);
			expect(result).to.be.instanceOf(Set);
			expect(result.size).to.equal(2);
		});
	});

	describe('similarityScore', () => {
		it('should return correct similarity score', () => {
			const set1 = new Set(['test', 'query']);
			const set2 = new Set(['query', 'another']);
			const result = jaccardSimilarity['similarityScore'](set1, set2);
			expect(result).to.be.closeTo(0.33, 0.01);
		});
	});

	it('should return the correct similarity score', () => {
		const path1 = 'folder1/folder2/file1';
		const set2 = new Set(['folder1', 'folder2', 'file2']);
		const expectedScore = 2 / 3; // Assuming Jaccard similarity score calculation
		const actualScore = jaccardSimilarity.pathSimilarity(path1, set2);
		expect(actualScore).to.equal(expectedScore);
	});
});