mirror of
https://github.com/OPM/ResInsight.git
synced 2026-10-25 06:38:13 -05:00
Kristian Bendiksen
ac43c9ac71
Cover the easily-reachable gaps in Project API test coverage: save() round trip, views()/view() lookup, well_path_by_name() hit and miss, the grid_case_groups()/grid_case_group(id) lookup pair, and the async key_values iterator.
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
import sys
|
|
import os
|
|
import pytest
|
|
|
|
sys.path.insert(1, os.path.join(sys.path[0], "../../"))
|
|
import rips
|
|
|
|
|
|
def test_set_and_get_values(rips_instance, initialize_test):
|
|
project = rips_instance.project
|
|
key = "my-key"
|
|
input_values = [float(x) for x in range(2000)]
|
|
project.set_key_values(key, input_values)
|
|
|
|
output_values = project.key_values(key)
|
|
assert input_values == output_values
|
|
|
|
|
|
def test_set_and_get_values_async(rips_instance, initialize_test):
|
|
project = rips_instance.project
|
|
key = "my-async-key"
|
|
input_values = [float(x) for x in range(2000)]
|
|
project.set_key_values(key, input_values)
|
|
|
|
accumulated = []
|
|
chunk_count = 0
|
|
for chunk in project.key_values_async(key):
|
|
accumulated.extend(chunk.values)
|
|
chunk_count += 1
|
|
assert chunk_count > 0
|
|
assert input_values == accumulated
|
|
|
|
|
|
def test_get_non_existing_key(rips_instance, initialize_test):
|
|
project = rips_instance.project
|
|
key = "this-key-does-not-exist"
|
|
|
|
with pytest.raises(rips.RipsError, match="No matching key found."):
|
|
project.key_values(key)
|
|
|
|
|
|
def test_remove_non_existing_key(rips_instance, initialize_test):
|
|
project = rips_instance.project
|
|
key = "this-key-does-not-exist"
|
|
|
|
with pytest.raises(rips.RipsError, match="No matching key found."):
|
|
project.remove_key_values(key)
|
|
|
|
|
|
def test_set_and_remove_and_get_values(rips_instance, initialize_test):
|
|
project = rips_instance.project
|
|
key = "my-key"
|
|
input_values = [float(x) for x in range(2000)]
|
|
project.set_key_values(key, input_values)
|
|
|
|
output_values = project.key_values(key)
|
|
assert input_values == output_values
|
|
|
|
project.remove_key_values(key)
|
|
|
|
with pytest.raises(rips.RipsError, match="No matching key found."):
|
|
project.key_values(key)
|