Exemple #1
0
def get_kraken_calls_test():
    for md in ["bike", "walking", "bss", "car"]:
        req = {"origin_mode": [md], "destination_mode": [md]}
        assert get_kraken_calls(req) == [(md, md)]

    req = {"origin_mode": ["bss", "walking"], "destination_mode": ["walking"]}
    assert get_kraken_calls(req) == [("walking", "walking")]

    req = {"origin_mode": ["bike", "walking"], "destination_mode": ["walking"]}
    assert get_kraken_calls(req) == [("walking", "walking"),
                                     ("bike", "walking")]

    req = {"origin_mode": ["bike", "walking"], "destination_mode": ["bss"]}
    assert get_kraken_calls(req) == [("bike", "bss")]

    req = {"origin_mode": ["bike", "car"], "destination_mode": ["bss"]}
    assert get_kraken_calls(req) == [("bike", "bss"), ("car", "bss")]
Exemple #2
0
def get_kraken_calls_test():
    for md in ["bike", "walking", "bss", "car", "ridesharing", "taxi"]:
        req = {"origin_mode": [md], "destination_mode": [md]}
        assert get_kraken_calls(req) == {(md, md)}

    req = {"origin_mode": ["bss", "walking"], "destination_mode": ["walking"]}
    assert get_kraken_calls(req) == {("walking", "walking")}

    req = {"origin_mode": ["bike", "walking"], "destination_mode": ["walking"]}
    assert get_kraken_calls(req) == {("walking", "walking"),
                                     ("bike", "walking")}

    req = {"origin_mode": ["bike", "walking"], "destination_mode": ["bss"]}
    assert get_kraken_calls(req) == {("bike", "bss")}

    req = {"origin_mode": ["bike", "car"], "destination_mode": ["bss"]}
    assert get_kraken_calls(req) == {("bike", "bss"), ("car", "bss")}

    req = {"origin_mode": ["bss", "bike"], "destination_mode": ["bike"]}
    assert get_kraken_calls(req) == {("bike", "bike")}

    req = {
        "origin_mode": ["taxi", "bike"],
        "destination_mode": ["walking", "bss", "car", "ridesharing", "taxi"]
    }
    assert get_kraken_calls(req) == {
        ('taxi', 'taxi'),
        ('taxi', 'bss'),
        ('bike', 'bss'),
        ('bike', 'walking'),
        ('bike', 'taxi'),
        ('taxi', 'walking'),
        ('taxi', 'ridesharing'),
    }

    req = {
        "origin_mode": ["ridesharing", "taxi"],
        "destination_mode":
        ["walking", "bss", "bike", "car", "ridesharing", "taxi"],
    }
    assert get_kraken_calls(req) == {
        ('taxi', 'taxi'),
        ('ridesharing', 'walking'),
        ('taxi', 'bss'),
        ('ridesharing', 'bss'),
        ('taxi', 'ridesharing'),
        ('taxi', 'walking'),
        ('ridesharing', 'taxi'),
    }
def get_kraken_calls_invalid_2_test():
    with pytest.raises(HTTPException):
        get_kraken_calls({"origin_mode": ["bss", "walking"], "destination_mode": ["bike"]})
def get_kraken_calls_test():
    for md in ["bike", "walking", "bss", "car", "ridesharing", "taxi"]:
        req = {"origin_mode": [md], "destination_mode": [md]}
        assert get_kraken_calls(req) == {(md, md, "indifferent")}

    req = {"origin_mode": ["bss", "walking"], "destination_mode": ["walking"], "direct_path": "none"}
    assert get_kraken_calls(req) == {("walking", "walking", "none")}

    req = {"origin_mode": ["bike", "walking"], "destination_mode": ["walking"], "direct_path": "indifferent"}
    assert get_kraken_calls(req) == {("walking", "walking", "indifferent"), ("bike", "walking", "indifferent")}

    req = {"origin_mode": ["bike", "walking"], "destination_mode": ["bss"], "direct_path": "indifferent"}
    assert get_kraken_calls(req) == {("bike", "bss", "indifferent")}

    req = {"origin_mode": ["bike", "car"], "destination_mode": ["bss"]}
    assert get_kraken_calls(req) == {("bike", "bss", "indifferent"), ("car", "bss", "indifferent")}

    req = {"origin_mode": ["bss", "bike"], "destination_mode": ["bike"]}
    assert get_kraken_calls(req) == {("bike", "bike", "indifferent")}

    req = {"origin_mode": ["taxi", "bike"], "destination_mode": ["walking", "bss", "car", "ridesharing", "taxi"]}
    assert get_kraken_calls(req) == {
        ('taxi', 'taxi', "indifferent"),
        ('taxi', 'bss', "indifferent"),
        ('bike', 'bss', "indifferent"),
        ('bike', 'walking', "indifferent"),
        ('bike', 'taxi', "indifferent"),
        ('taxi', 'walking', "indifferent"),
        ('taxi', 'ridesharing', "indifferent"),
    }

    req = {
        "origin_mode": ["ridesharing", "taxi"],
        "destination_mode": ["walking", "bss", "bike", "car", "ridesharing", "taxi"],
    }
    assert get_kraken_calls(req) == {
        ('taxi', 'taxi', "indifferent"),
        ('ridesharing', 'walking', "indifferent"),
        ('taxi', 'bss', "indifferent"),
        ('ridesharing', 'bss', "indifferent"),
        ('taxi', 'ridesharing', "indifferent"),
        ('taxi', 'walking', "indifferent"),
        ('ridesharing', 'taxi', "indifferent"),
    }

    req = {"origin_mode": ["bss", "walking"], "destination_mode": ["walking"], "direct_path_mode": ["walking"]}
    assert get_kraken_calls(req) == {("walking", "walking", "indifferent")}

    req = {"origin_mode": ["bss", "walking"], "destination_mode": ["walking"], "direct_path_mode": ["bike"]}
    assert get_kraken_calls(req) == {("walking", "walking", "indifferent"), ("bike", "bike", "only")}

    req = {
        "origin_mode": ["bss", "walking"],
        "destination_mode": ["walking"],
        "direct_path_mode": ["bike"],
        "direct_path": "none",
    }
    assert get_kraken_calls(req) == {("walking", "walking", "none")}