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);
}
