From 6a74572269964d4d9dbfa5dc47a3f27a18a572a4 Mon Sep 17 00:00:00 2001 From: Jakub Kuszneruk Date: Wed, 7 Sep 2022 12:45:16 +0200 Subject: [PATCH] Fix e2e tests for filtering columns (#994) --- e2e_tests/standard/test_base.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/e2e_tests/standard/test_base.py b/e2e_tests/standard/test_base.py index caaff48ee..56bbda99b 100644 --- a/e2e_tests/standard/test_base.py +++ b/e2e_tests/standard/test_base.py @@ -250,7 +250,7 @@ def _test_fetch_from_container(self, init_container, get_containers_as_rows): assert container1.get_attribute_value(key2) == value2 assert container2.get_attribute_value(key1) == value1 with pytest.raises(ValueError): - assert container2.get_attribute_value(key2) + container2.get_attribute_value(key2) def get_container1(**kwargs): containers_as_rows = get_containers_as_rows(**kwargs) @@ -269,15 +269,19 @@ def get_container1(**kwargs): assert columns_none.get_attribute_value(key2) == value2 columns_empty = get_container1(columns=[]) - assert columns_empty.get_attribute_value(key1) is None - assert columns_empty.get_attribute_value(key2) is None + with pytest.raises(ValueError): + columns_empty.get_attribute_value(key1) + with pytest.raises(ValueError): + columns_empty.get_attribute_value(key2) columns_with_one_key = get_container1(columns=[key1]) assert columns_with_one_key.get_attribute_value(key1) == value1 - assert columns_with_one_key.get_attribute_value(key2) is None + with pytest.raises(ValueError): + columns_with_one_key.get_attribute_value(key2) columns_with_one_key = get_container1(columns=[key2]) - assert columns_with_one_key.get_attribute_value(key1) is None + with pytest.raises(ValueError): + columns_with_one_key.get_attribute_value(key1) assert columns_with_one_key.get_attribute_value(key2) == value2 def test_fetch_runs_table(self, environment):