Start of tests

Author: adamtheturtle

tests/test_query.py

@@ -196,6 +196,8 @@ def test_default(
             image=high_quality_image,
             max_num_results=2,
         )
+        assert 'target_data' in top_match
+        assert 'target_data' not in second_match
 
     def test_top(
         self,
@@ -219,6 +221,8 @@ def test_top(
             image=high_quality_image,
             max_num_results=2,
         )
+        assert 'target_data' in top_match
+        assert 'target_data' not in second_match
 
     def test_none(
         self,
@@ -242,6 +246,8 @@ def test_none(
             image=high_quality_image,
             max_num_results=2,
         )
+        assert 'target_data' not in top_match
+        assert 'target_data' not in second_match
 
     def test_all(
         self,
@@ -265,3 +271,5 @@ def test_all(
             image=high_quality_image,
             max_num_results=2,
         )
+        assert 'target_data' in top_match
+        assert 'target_data' in second_match