7fa972be6a
version: sunnypilot v2025.003.000 (dev) date: 2026-02-09T02:04:38 master commit: 254f55ac15a40343d7255f2f098de3442e0c4a6f
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() |