def test_false_positive_low_confidence(self):
assert calculate_tag([
{
'label': 'false-positive',
'confidence': 0.39
},
]) == (FALSE_POSITIVE, 0.85)
def test_low_confidence_animal_and_false_positive(self):
assert calculate_tag([
{
'label': 'false-positive',
'confidence': 0.46
},
{
'label': 'rat',
'confidence': 0.7
},
]) == (UNIDENTIFIED, 0.85)
def test_animal_and_false_positive(self):
assert calculate_tag([
{
'label': 'false-positive',
'confidence': 0.9
},
{
'label': 'rat',
'confidence': 0.88
},
]) == ('rat', 0.88)
def test_multi_track_two_animals(self):
assert calculate_tag([
{
'label': 'rat',
'confidence': 0.9
},
{
'label': 'possum',
'confidence': 0.9
},
]) == (UNIDENTIFIED, 0.85)
def test_multi_track_same_animal(self):
assert calculate_tag([
{
'label': 'rat',
'confidence': 0.5
},
{
'label': 'rat',
'confidence': 0.9
},
]) == ('rat', 0.9)
def test_many_strong(self):
assert calculate_tag([
{
'label': 'rat',
'confidence': 0.9
},
{
'label': 'possum',
'confidence': 0.9
},
{
'label': 'false-positive',
'confidence': 0.9
},
]) == (UNIDENTIFIED, 0.85)
def test_multi_false_positive(self):
assert calculate_tag([
{
'label': 'false-positive',
'confidence': 0.39
},
{
'label': 'false-positive',
'confidence': 0.50
},
{
'label': 'false-positive',
'confidence': 0.12
},
]) == (FALSE_POSITIVE, 0.85)
def test_one_track(self):
assert calculate_tag([{
'label': 'rat',
'confidence': 0.9,
}]) == ('rat', 0.9)
def test_no_tracks(self):
assert calculate_tag([]) == (FALSE_POSITIVE, 0.85)
def test_one_track_low_confidence(self):
assert calculate_tag([{
'label': 'rat',
'confidence': 0.5,
}]) == (UNIDENTIFIED, 0.85)