public void HammingSimilarityIsSummedUpAccrossAllSubFingerprintsTest() { long[] buckets = new long[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }; var queryHash = new HashData(new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 11 }, buckets); const int DefaultThreshold = 5; const int FirstTrackId = 20; const int FirstSubFingerprintId = 10; const int SecondSubFingerprintId = 11; var firstTrackReference = new ModelReference<int>(FirstTrackId); SubFingerprintData firstResult = new SubFingerprintData(new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }, new ModelReference<int>(FirstSubFingerprintId), firstTrackReference); SubFingerprintData secondResult = new SubFingerprintData(new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 10, 12 }, new ModelReference<int>(SecondSubFingerprintId), firstTrackReference); modelService.Setup(service => service.ReadSubFingerprintDataByHashBucketsWithThreshold(buckets, DefaultThreshold)) .Returns(new List<SubFingerprintData> { firstResult, secondResult }); modelService.Setup(service => service.ReadTrackByReference(firstTrackReference)) .Returns(new TrackData { ISRC = "isrc", TrackReference = firstTrackReference }); var queryResult = queryFingerprintService.Query(modelService.Object, new List<HashData> { queryHash }, new DefaultQueryConfiguration()); Assert.IsTrue(queryResult.IsSuccessful); Assert.AreEqual("isrc", queryResult.BestMatch.Track.ISRC); Assert.AreEqual(firstTrackReference, queryResult.BestMatch.Track.TrackReference); Assert.AreEqual(9 + 8, queryResult.BestMatch.Similarity); Assert.AreEqual(1, queryResult.AnalyzedCandidatesCount); Assert.AreEqual(1, queryResult.ResultEntries.Count); }
private IEnumerable<SubFingerprintData> GetSubFingerprints(IModelService modelService, HashData hash, IQueryConfiguration queryConfiguration) { if (!string.IsNullOrEmpty(queryConfiguration.TrackGroupId)) { return modelService.ReadSubFingerprintDataByHashBucketsThresholdWithGroupId(hash.HashBins, queryConfiguration.ThresholdVotes, queryConfiguration.TrackGroupId); } return modelService.ReadSubFingerprintDataByHashBucketsWithThreshold(hash.HashBins, queryConfiguration.ThresholdVotes); }
public void MaximumNumberOfReturnedTracksIsLessThanAnalyzedCandidatesResultsTest() { long[] buckets = new long[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }; var queryHash = new HashData(new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 11 }, buckets); const int DefaultThreshold = 5; const int FirstTrackId = 20; const int SecondTrackId = 21; const int ThirdTrackId = 22; const int FirstSubFingerprintId = 10; const int SecondSubFingerprintId = 11; var firstTrackReference = new ModelReference<int>(FirstTrackId); var thirdTrackReference = new ModelReference<int>(ThirdTrackId); SubFingerprintData firstResult = new SubFingerprintData( new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }, new ModelReference<int>(FirstSubFingerprintId), firstTrackReference); SubFingerprintData secondResult = new SubFingerprintData( new byte[] { 11, 2, 13, 4, 15, 6, 7, 8, 10, 12 }, new ModelReference<int>(SecondSubFingerprintId), new ModelReference<int>(SecondTrackId)); SubFingerprintData thirdResult = new SubFingerprintData( new byte[] { 1, 2, 3, 4, 5, 15, 7, 8, 10, 12 }, new ModelReference<int>(SecondSubFingerprintId), new ModelReference<int>(ThirdTrackId)); modelService.Setup( service => service.ReadSubFingerprintDataByHashBucketsWithThreshold(buckets, DefaultThreshold)).Returns( new List<SubFingerprintData> { firstResult, secondResult, thirdResult }); modelService.Setup(service => service.ReadTrackByReference(firstTrackReference)).Returns( new TrackData { ISRC = "isrc", TrackReference = firstTrackReference }); modelService.Setup(service => service.ReadTrackByReference(thirdTrackReference)).Returns( new TrackData { ISRC = "isrc_2", TrackReference = thirdTrackReference }); var queryResult = queryFingerprintService.Query( modelService.Object, new List<HashData> { queryHash }, new CustomQueryConfiguration { MaximumNumberOfTracksToReturnAsResult = 2, ThresholdVotes = DefaultThreshold }); Assert.IsTrue(queryResult.IsSuccessful); Assert.AreEqual("isrc", queryResult.BestMatch.Track.ISRC); Assert.AreEqual(firstTrackReference, queryResult.BestMatch.Track.TrackReference); Assert.AreEqual(9, queryResult.BestMatch.Similarity); Assert.AreEqual(3, queryResult.AnalyzedCandidatesCount); Assert.AreEqual(2, queryResult.ResultEntries.Count); Assert.AreEqual(firstTrackReference, queryResult.ResultEntries[0].Track.TrackReference); Assert.AreEqual(thirdTrackReference, queryResult.ResultEntries[1].Track.TrackReference); }
public void DeleteTrackTest() { const int Threshold = 5; TrackData track = new TrackData("isrc", "artist", "title", "album", 1986, 200); var trackReference = ModelService.InsertTrack(track); var hashData = new HashData(GenericSignature, GenericHashBuckets); ModelService.InsertHashDataForTrack(new[] { hashData }, trackReference); ModelService.DeleteTrack(trackReference); var subFingerprints = ModelService.ReadSubFingerprintDataByHashBucketsWithThreshold(GenericHashBuckets, Threshold); Assert.IsTrue(subFingerprints.Any() == false); TrackData actualTrack = ModelService.ReadTrackByReference(trackReference); Assert.IsNull(actualTrack); }
public void InsertHashDataTest() { const int Threshold = 5; TrackData expectedTrack = new TrackData("isrc", "artist", "title", "album", 1986, 200); var trackReference = ModelService.InsertTrack(expectedTrack); var hashData = new HashData(GenericSignature, GenericHashBuckets); ModelService.InsertHashDataForTrack(new[] { hashData }, trackReference); var subFingerprints = ModelService.ReadSubFingerprintDataByHashBucketsWithThreshold(GenericHashBuckets, Threshold); Assert.IsTrue(subFingerprints.Count == 1); Assert.AreEqual(trackReference, subFingerprints[0].TrackReference); Assert.IsFalse(subFingerprints[0].SubFingerprintReference.GetHashCode() == 0); for (int i = 0; i < GenericSignature.Length; i++) { Assert.AreEqual(GenericSignature[i], subFingerprints[0].Signature[i]); } }
public void OnlyTracksWithGroupIdAreConsideredAsPotentialCandidatesTest() { long[] buckets = new long[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }; var queryHash = new HashData(new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 11 }, buckets); const int DefaultThreshold = 5; const int FirstTrackId = 20; const int FirstSubFingerprintId = 10; var firstTrackReference = new ModelReference<int>(FirstTrackId); SubFingerprintData firstResult = new SubFingerprintData( new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }, new ModelReference<int>(FirstSubFingerprintId), firstTrackReference); modelService.Setup( service => service.ReadSubFingerprintDataByHashBucketsThresholdWithGroupId(buckets, DefaultThreshold, "group-id")) .Returns(new List<SubFingerprintData> { firstResult }); modelService.Setup(service => service.ReadTrackByReference(firstTrackReference)).Returns( new TrackData { ISRC = "isrc", TrackReference = firstTrackReference }); var queryResult = queryFingerprintService.Query( modelService.Object, new List<HashData> { queryHash }, new CustomQueryConfiguration { TrackGroupId = "group-id" }); Assert.IsTrue(queryResult.IsSuccessful); Assert.AreEqual("isrc", queryResult.BestMatch.Track.ISRC); Assert.AreEqual(firstTrackReference, queryResult.BestMatch.Track.TrackReference); }
public void NoResultsReturnedFromUnderlyingStorageTest() { var queryHash = new HashData(new byte[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 11 }, new long[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }); modelService.Setup(service => service.ReadSubFingerprintDataByHashBucketsWithThreshold(It.IsAny<long[]>(), 10)).Returns(new List<SubFingerprintData>()); var queryResult = queryFingerprintService.Query( modelService.Object, new List<HashData> { queryHash }, new CustomQueryConfiguration { MaximumNumberOfTracksToReturnAsResult = 1, ThresholdVotes = 10 }); Assert.IsFalse(queryResult.IsSuccessful); Assert.IsNull(queryResult.BestMatch); Assert.AreEqual(0, queryResult.AnalyzedCandidatesCount); Assert.AreEqual(0, queryResult.ResultEntries.Count); }
public void ReadSubFingerprintsByHashBucketsHavingThresholdWithGroupIdTest() { const int Threshold = 5; TrackData firstTrack = new TrackData("isrc1", "artist", "title", "album", 1986, 200) { GroupId = "first-group-id" }; var firstTrackReference = ModelService.InsertTrack(firstTrack); TrackData secondTrack = new TrackData("isrc2", "artist", "title", "album", 1986, 200) { GroupId = "second-group-id" }; var secondTrackReference = ModelService.InsertTrack(secondTrack); Assert.IsFalse(firstTrackReference.Equals(secondTrackReference)); long[] firstTrackBuckets = new long[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25 }; long[] secondTrackBuckets = new long[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25 }; var firstHashData = new HashData(GenericSignature, firstTrackBuckets); var secondHashData = new HashData(GenericSignature, secondTrackBuckets); ModelService.InsertHashDataForTrack(new[] { firstHashData }, firstTrackReference); ModelService.InsertHashDataForTrack(new[] { secondHashData }, secondTrackReference); // query buckets are similar with 5 elements from first track and 4 elements from second track long[] queryBuckets = new long[] { 3, 2, 5, 6, 7, 8, 7, 10, 11, 12, 13, 14, 15, 14, 17, 18, 19, 20, 21, 20, 23, 24, 25, 26, 25 }; var subFingerprints = ModelService.ReadSubFingerprintDataByHashBucketsThresholdWithGroupId(queryBuckets, Threshold, "first-group-id"); Assert.IsTrue(subFingerprints.Count == 1); Assert.AreEqual(firstTrackReference, subFingerprints[0].TrackReference); }