File: test_data_paths.py

package info (click to toggle)
fiona 1.10.1-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,632 kB
  • sloc: python: 12,616; makefile: 214; sh: 45
file content (57 lines) | stat: -rw-r--r-- 1,861 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
"""Tests of GDAL and PROJ data finding"""

import os.path

from click.testing import CliRunner
import pytest

import fiona
from fiona._env import GDALDataFinder, PROJDataFinder
from fiona.fio.main import main_group


@pytest.mark.wheel
def test_gdal_data_wheel():
    """Get GDAL data path from a wheel"""
    assert GDALDataFinder().search() == os.path.join(os.path.dirname(fiona.__file__), 'gdal_data')


@pytest.mark.wheel
def test_proj_data_wheel():
    """Get GDAL data path from a wheel"""
    assert PROJDataFinder().search() == os.path.join(os.path.dirname(fiona.__file__), 'proj_data')


@pytest.mark.wheel
def test_env_gdal_data_wheel():
    runner = CliRunner()
    result = runner.invoke(main_group, ['env', '--gdal-data'])
    assert result.exit_code == 0
    assert result.output.strip() == os.path.join(os.path.dirname(fiona.__file__), 'gdal_data')


@pytest.mark.wheel
def test_env_proj_data_wheel():
    runner = CliRunner()
    result = runner.invoke(main_group, ['env', '--proj-data'])
    assert result.exit_code == 0
    assert result.output.strip() == os.path.join(os.path.dirname(fiona.__file__), 'proj_data')


def test_env_gdal_data_environ(monkeypatch):
    monkeypatch.setenv('GDAL_DATA', '/foo/bar')
    runner = CliRunner()
    result = runner.invoke(main_group, ['env', '--gdal-data'])
    assert result.exit_code == 0
    assert result.output.strip() == '/foo/bar'


@pytest.mark.parametrize("data_directory_env", ["PROJ_LIB", "PROJ_DATA"])
def test_env_proj_data_environ(data_directory_env, monkeypatch):
    monkeypatch.delenv('PROJ_DATA', raising=False)
    monkeypatch.delenv('PROJ_LIB', raising=False)
    monkeypatch.setenv(data_directory_env, '/foo/bar')
    runner = CliRunner()
    result = runner.invoke(main_group, ['env', '--proj-data'])
    assert result.exit_code == 0
    assert result.output.strip() == '/foo/bar'