Skip to content

Commit e2a8a7d

Browse files
committed
improve tests
1 parent b91359b commit e2a8a7d

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

tests/unittests/tools/test_gemini_schema_util.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,9 @@ def test_to_gemini_schema_array_string_types(self):
6565
"nonnullable_string": {"type": ["string"]},
6666
"nullable_string": {"type": ["string", "null"]},
6767
"nullable_number": {"type": ["null", "integer"]},
68-
"object_nullable": {"type": ["object", "null"]},
68+
"nullable_object": {"type": ["object", "null"]},
6969
"multi_types_nullable": {"type": ["string", "null", "integer"]},
70+
"only_null": {"type": "null"},
7071
"empty_default_object": {},
7172
},
7273
}
@@ -84,14 +85,17 @@ def test_to_gemini_schema_array_string_types(self):
8485
assert gemini_schema.properties["nullable_number"].type == Type.INTEGER
8586
assert gemini_schema.properties["nullable_number"].nullable
8687

87-
assert gemini_schema.properties["object_nullable"].type == Type.OBJECT
88-
assert gemini_schema.properties["object_nullable"].nullable
88+
assert gemini_schema.properties["nullable_object"].type == Type.OBJECT
89+
assert gemini_schema.properties["nullable_object"].nullable
8990

9091
assert gemini_schema.properties["multi_types_nullable"].any_of == [
9192
Schema(type=Type.STRING),
9293
Schema(type=Type.INTEGER),
9394
]
9495
assert gemini_schema.properties["multi_types_nullable"].nullable
96+
97+
assert gemini_schema.properties["only_null"].type is None
98+
assert gemini_schema.properties["only_null"].nullable
9599

96100
assert gemini_schema.properties["empty_default_object"].type == Type.OBJECT
97101
assert gemini_schema.properties["empty_default_object"].nullable is None

0 commit comments

Comments
 (0)