Spaces:
Running
Running
| # Copyright (c) Facebook, Inc. and its affiliates. | |
| import random | |
| import unittest | |
| from densepose.data.video import FirstKFramesSelector, LastKFramesSelector, RandomKFramesSelector | |
| class TestFrameSelector(unittest.TestCase): | |
| def test_frame_selector_random_k_1(self): | |
| _SEED = 43 | |
| _K = 4 | |
| random.seed(_SEED) | |
| selector = RandomKFramesSelector(_K) | |
| frame_tss = list(range(0, 20, 2)) | |
| _SELECTED_GT = [0, 8, 4, 6] | |
| selected = selector(frame_tss) | |
| self.assertEqual(_SELECTED_GT, selected) | |
| def test_frame_selector_random_k_2(self): | |
| _SEED = 43 | |
| _K = 10 | |
| random.seed(_SEED) | |
| selector = RandomKFramesSelector(_K) | |
| frame_tss = list(range(0, 6, 2)) | |
| _SELECTED_GT = [0, 2, 4] | |
| selected = selector(frame_tss) | |
| self.assertEqual(_SELECTED_GT, selected) | |
| def test_frame_selector_first_k_1(self): | |
| _K = 4 | |
| selector = FirstKFramesSelector(_K) | |
| frame_tss = list(range(0, 20, 2)) | |
| _SELECTED_GT = frame_tss[:_K] | |
| selected = selector(frame_tss) | |
| self.assertEqual(_SELECTED_GT, selected) | |
| def test_frame_selector_first_k_2(self): | |
| _K = 10 | |
| selector = FirstKFramesSelector(_K) | |
| frame_tss = list(range(0, 6, 2)) | |
| _SELECTED_GT = frame_tss[:_K] | |
| selected = selector(frame_tss) | |
| self.assertEqual(_SELECTED_GT, selected) | |
| def test_frame_selector_last_k_1(self): | |
| _K = 4 | |
| selector = LastKFramesSelector(_K) | |
| frame_tss = list(range(0, 20, 2)) | |
| _SELECTED_GT = frame_tss[-_K:] | |
| selected = selector(frame_tss) | |
| self.assertEqual(_SELECTED_GT, selected) | |
| def test_frame_selector_last_k_2(self): | |
| _K = 10 | |
| selector = LastKFramesSelector(_K) | |
| frame_tss = list(range(0, 6, 2)) | |
| _SELECTED_GT = frame_tss[-_K:] | |
| selected = selector(frame_tss) | |
| self.assertEqual(_SELECTED_GT, selected) | |