def test_get_loader_config_override(): """Test two loaders that override the defaults.""" configs = _get_loader_configs(TEST_CONFIG) # Check each config overrode the defaults. assert len(configs) == 2 assert configs[0]['num_workers'] == 10 assert configs[3]['num_workers'] == 5 assert configs[0]['delay_queue_ms'] == 100 assert configs[3]['delay_queue_ms'] == 0 # Check the defaults are still there. assert configs[0]['force_synchronous'] is False assert configs[3]['force_synchronous'] is False
def test_get_loader_config_defaults(): """Test config that has defaults but no octree loaders.""" config = { "loader_defaults": { "force_synchronous": False, "num_workers": 10, "delay_queue_ms": 0, }, "octree": {}, } configs = _get_loader_configs(config) assert len(configs) == 1 configs[0]['num_workers'] == 10 configs[0]['delay_queue_ms'] == 10
def test_get_loader_config_error(): """Test that defaults are required.""" config = {} with pytest.raises(KeyError): _get_loader_configs(config)