mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-07-22 21:32:13 +08:00
Vehicle Researcher
161005a0c1
date: 2025-11-19T19:10:11 master commit: a29fdbd02407d41ecbcc69d151bb4837bfba3cbc
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() |