Skip to content

Commit 0493d04

Browse files
committed
Add max_new_tokens to GenerationConfig tests
1 parent 0e787d1 commit 0493d04

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/cpp/continuous_batching/src/tests/generate_config.cpp

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,23 @@
77

88
TEST(GenerationConfigTest, invalid_temperature) {
99
ov::genai::GenerationConfig config;
10+
config.max_new_tokens = 20;
1011
config.temperature = -0.1;
1112
config.do_sample = true;
1213
EXPECT_THROW(config.validate(), ov::Exception);
1314
}
1415

1516
TEST(GenerationConfigTest, valid_temperature) {
1617
ov::genai::GenerationConfig config;
18+
config.max_new_tokens = 20;
1719
config.do_sample = true;
1820
config.temperature = 0.1;
1921
EXPECT_NO_THROW(config.validate());
2022
}
2123

2224
TEST(GenerationConfigTest, invalid_top_p) {
2325
ov::genai::GenerationConfig config;
26+
config.max_new_tokens = 20;
2427
config.do_sample = true;
2528
config.top_p = -0.5;
2629
EXPECT_THROW(config.validate(), ov::Exception);
@@ -30,13 +33,15 @@ TEST(GenerationConfigTest, invalid_top_p) {
3033

3134
TEST(GenerationConfigTest, valid_top_p) {
3235
ov::genai::GenerationConfig config;
36+
config.max_new_tokens = 20;
3337
config.do_sample = true;
3438
config.top_p = 0.1;
3539
EXPECT_NO_THROW(config.validate());
3640
}
3741

3842
TEST(GenerationConfigTest, invalid_repeatition_penalty) {
3943
ov::genai::GenerationConfig config;
44+
config.max_new_tokens = 20;
4045
config.do_sample = true;
4146
config.repetition_penalty = -3.0;
4247
EXPECT_THROW(config.validate(), ov::Exception);
@@ -46,6 +51,7 @@ TEST(GenerationConfigTest, invalid_repeatition_penalty) {
4651

4752
TEST(GenerationConfigTest, valid_repeatition_penalty) {
4853
ov::genai::GenerationConfig config;
54+
config.max_new_tokens = 20;
4955
config.do_sample = true;
5056
config.repetition_penalty = 1.8;
5157
EXPECT_NO_THROW(config.validate());
@@ -55,6 +61,7 @@ TEST(GenerationConfigTest, valid_repeatition_penalty) {
5561

5662
TEST(GenerationConfigTest, invalid_presence_penalty) {
5763
ov::genai::GenerationConfig config;
64+
config.max_new_tokens = 20;
5865
config.do_sample = true;
5966
config.presence_penalty = 3.0;
6067
EXPECT_THROW(config.validate(), ov::Exception);
@@ -64,6 +71,7 @@ TEST(GenerationConfigTest, invalid_presence_penalty) {
6471

6572
TEST(GenerationConfigTest, valid_presence_penalty) {
6673
ov::genai::GenerationConfig config;
74+
config.max_new_tokens = 20;
6775
config.do_sample = true;
6876
config.presence_penalty = 1.8;
6977
EXPECT_NO_THROW(config.validate());
@@ -73,6 +81,7 @@ TEST(GenerationConfigTest, valid_presence_penalty) {
7381

7482
TEST(GenerationConfigTest, invalid_frequency_penalty) {
7583
ov::genai::GenerationConfig config;
84+
config.max_new_tokens = 20;
7685
config.do_sample = true;
7786
config.frequency_penalty = 3.0;
7887
EXPECT_THROW(config.validate(), ov::Exception);
@@ -82,6 +91,7 @@ TEST(GenerationConfigTest, invalid_frequency_penalty) {
8291

8392
TEST(GenerationConfigTest, valid_frequency_penalty) {
8493
ov::genai::GenerationConfig config;
94+
config.max_new_tokens = 20;
8595
config.do_sample = true;
8696
config.frequency_penalty = 1.8;
8797
EXPECT_NO_THROW(config.validate());

0 commit comments

Comments
 (0)