Skip to content

Commit 9402ec6

Browse files
committed
fix: fixed TrainOnWarAndPeaceWordLevel test
1 parent 6d567e4 commit 9402ec6

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

tests/markov_chain_tests.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,8 @@ TEST(MarkovTextModelTest, CharacterLevelGeneration) {
7272
TEST(MarkovTextModelTest, TrainOnWarAndPeaceWordLevel) {
7373
using namespace ptm;
7474

75-
std::ifstream in("tests/data/war_and_peace.txt");
76-
ASSERT_TRUE(in.good()) << "Не удалось открыть файл tests/data/war_and_peace_ru.txt";
75+
std::ifstream in("../../tests/war_and_peace.txt");
76+
ASSERT_TRUE(in.good()) << "Не удалось открыть файл ../../tests/war_and_peace.txt";
7777

7878
std::stringstream buffer;
7979
buffer << in.rdbuf();
@@ -90,13 +90,13 @@ TEST(MarkovTextModelTest, TrainOnWarAndPeaceWordLevel) {
9090

9191
auto has_token = [&](const std::string& token) { return std::ranges::find(states, token) != states.end(); };
9292

93-
EXPECT_TRUE(has_token("и")) << "Слово \"и\" не найдено в словаре";
94-
EXPECT_TRUE(has_token("в")) << "Слово \"в\" не найдено в словаре";
95-
EXPECT_TRUE(has_token("на")) << "Слово \"на\" не найдено в словаре";
93+
EXPECT_TRUE(has_token("and")) << "Слово \"and\" не найдено в словаре";
94+
EXPECT_TRUE(has_token("in")) << "Слово \"in\" не найдено в словаре";
95+
EXPECT_TRUE(has_token("on")) << "Слово \"on\" не найдено в словаре";
9696

9797
std::mt19937 rng(123);
9898

99-
std::string generated = model.GenerateText(50, rng, "и");
99+
std::string generated = model.GenerateText(50, rng, "and");
100100
EXPECT_FALSE(generated.empty()) << "Сгенерированный текст пустой";
101101

102102
std::size_t space_count = std::ranges::count(generated, ' ');

0 commit comments

Comments
 (0)