From e7a9bae60107576c147d5eb4ab19bf4fba518b38 Mon Sep 17 00:00:00 2001
From: Rob Hallam <0504004h@student.gla.ac.uk>
Date: Mon, 16 Sep 2024 01:43:48 +0100
Subject: [PATCH] test: VAFE _keep_num test

---
 test/test_feature_extractors.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/test/test_feature_extractors.py b/test/test_feature_extractors.py
index f8de329..5286a31 100644
--- a/test/test_feature_extractors.py
+++ b/test/test_feature_extractors.py
@@ -310,7 +310,19 @@ class TestVideoActivityFeatureExtractor(unittest.TestCase):
         test_extractor.teardown()
         self.assertTrue(test_extractor.features)
 
-    # TODO: add sample video with activity to test _activitydetect()
+    def test_keep_num(self):
+        """Test keep_num keeps 0"""
+        num_features = 1
+        min_duration = 0
+
+        video_source = TestSourceMedia().one_colour_silent_audio()
+        test_extractor = extractors.VideoActivityFeatureExtractor(input_files=video_source,
+                                                                  num_features=num_features,
+                                                                  min_duration=min_duration)
+        test_extractor.setup()
+        test_extractor.run()
+        test_extractor.teardown()
+        self.assertEqual(len(test_extractor.features), num_features)
 
 class TestJSONFeatureExtractor(unittest.TestCase):
     """Test JSONFeatureExtractor"""