Esempi in Python per EventSource.from_config

Esempio n. 1

def test_allowed_tels_from_config():
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    config = Config({'EventSource': {
        'input_url': dataset,
        'allowed_tels': {1, 3}
    }})
    reader = EventSource.from_config(config=config, parent=None)
    assert len(reader.allowed_tels) == 2

Esempio n. 2

File: test_event_source_factory.py Progetto: dipierr/ctapipe

def test_factory_allowed_tels_from_config():
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    config = Config({'EventSource': {
        'input_url': dataset,
        'allowed_tels': {1, 3}
    }})
    reader = EventSource.from_config(config=config, parent=None)
    assert len(reader.allowed_tels) == 2

Esempio n. 3

def test_from_config_default():
    old_default = EventSource.input_url.default_value
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    EventSource.input_url.default_value = dataset
    config = Config()
    reader = EventSource.from_config(config=config, parent=None)
    assert isinstance(reader, SimTelEventSource)
    assert reader.input_url == dataset
    EventSource.input_url.default_value = old_default

Esempio n. 4

def test_max_events_from_config():
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    max_events = 10
    config = Config({'EventSource': {
        'input_url': dataset,
        'max_events': max_events,
    }})
    reader = EventSource.from_config(config=config)
    assert reader.max_events == max_events

Esempio n. 5

File: test_event_source_factory.py Progetto: dipierr/ctapipe

def test_factory_max_events_from_config():
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    max_events = 10
    config = Config({'EventSource': {
        'input_url': dataset,
        'max_events': max_events,
    }})
    reader = EventSource.from_config(config=config)
    assert reader.max_events == max_events

Esempio n. 6

File: test_event_source_factory.py Progetto: dipierr/ctapipe

def test_from_config_default():
    old_default = EventSource.input_url.default_value
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    EventSource.input_url.default_value = dataset
    config = Config()
    reader = EventSource.from_config(config=config, parent=None)
    assert isinstance(reader, SimTelEventSource)
    assert reader.input_url == dataset
    EventSource.input_url.default_value = old_default

Esempio n. 7

def test_max_events_from_config():
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    max_events = 10
    config = Config(
        {"EventSource": {
            "input_url": dataset,
            "max_events": max_events
        }})
    reader = EventSource.from_config(config=config)
    assert reader.max_events == max_events

Esempio n. 8

def test_from_config_invalid_type():
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    EventSource.input_url.default_value = dataset
    config = Config({'EventSource': {'input_url': 124}})
    with pytest.raises(TraitError):
        EventSource.from_config(config=config, parent=None)

Esempio n. 9

def test_from_config():
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    config = Config({'EventSource': {'input_url': dataset}})
    reader = EventSource.from_config(config=config, parent=None)
    assert isinstance(reader, SimTelEventSource)
    assert reader.input_url == dataset

Esempio n. 10

File: test_event_source_factory.py Progetto: dipierr/ctapipe

def test_from_config_invalid_type():
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    EventSource.input_url.default_value = dataset
    config = Config({'EventSource': {'input_url': 124}})
    with pytest.raises(TraitError):
        EventSource.from_config(config=config, parent=None)

Esempio n. 11

File: test_event_source_factory.py Progetto: dipierr/ctapipe

def test_from_config():
    dataset = get_dataset_path("gamma_test_large.simtel.gz")
    config = Config({'EventSource': {'input_url': dataset}})
    reader = EventSource.from_config(config=config, parent=None)
    assert isinstance(reader, SimTelEventSource)
    assert reader.input_url == dataset