mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-06-25 16:02:14 +08:00
Vehicle Researcher
dd778596b7
date: 2025-03-15T21:10:51 master commit: fb7b9c0f9420d228f03362970ebcfb7237095cf3
19 lines
785 B
Python
19 lines
785 B
Python
import unittest
|
|
|
|
from extra.optimization.helpers import load_worlds
|
|
|
|
class TestKernelDataset(unittest.TestCase):
|
|
def test_load_worlds_filters(self):
|
|
all_kernels = load_worlds(filter_reduce=False, filter_noimage=False, filter_novariable=False)
|
|
|
|
reduce_kernels = load_worlds(filter_reduce=True, filter_noimage=False, filter_novariable=False)
|
|
self.assertGreater(len(all_kernels), len(reduce_kernels))
|
|
|
|
image_kernels = load_worlds(filter_reduce=False, filter_noimage=True, filter_novariable=False)
|
|
self.assertGreater(len(all_kernels), len(image_kernels))
|
|
|
|
variable_kernels = load_worlds(filter_reduce=False, filter_noimage=False, filter_novariable=True)
|
|
self.assertGreater(len(all_kernels), len(variable_kernels))
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |