add dynarray case to size_in_bytes test

vyperlang · Apr 21, 2022 · 97dcc0a · 97dcc0a
1 parent ccf2e0b
commit 97dcc0a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/functional/context/types/test_size_in_bytes.py b/tests/functional/context/types/test_size_in_bytes.py
@@ -27,6 +27,16 @@ def test_array_value_types(build_node, type_str, location, length, size):
     assert type_definition.size_in_bytes == size
 
 
+@pytest.mark.parametrize("type_str", BASE_TYPES)
+@pytest.mark.parametrize("location", LOCATIONS)
+@pytest.mark.parametrize("length", range(1, 4))
+def test_base_types_as_arrays(build_node, type_str, location, length):
+    node = build_node(f"DynArray[{type_str}, {length}]")
+    type_definition = get_type_from_annotation(node, location)
+
+    assert type_definition.size_in_bytes == 32 + length * 32
+
+
 @pytest.mark.parametrize("type_str", BASE_TYPES)
 @pytest.mark.parametrize("location", LOCATIONS)
 @pytest.mark.parametrize("length", range(1, 4))