diff options
Diffstat (limited to 'src/sentencepiece_processor_test.cc')
-rw-r--r-- | src/sentencepiece_processor_test.cc | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/sentencepiece_processor_test.cc b/src/sentencepiece_processor_test.cc index 09d6e1e..24ae595 100644 --- a/src/sentencepiece_processor_test.cc +++ b/src/sentencepiece_processor_test.cc @@ -18,6 +18,7 @@ #include <utility> #include "builder.h" +#include "filesystem.h" #include "model_interface.h" #include "normalizer.h" #include "sentencepiece.pb.h" @@ -631,8 +632,8 @@ TEST(SentencePieceProcessorTest, EndToEndTest) { test::ScopedTempFile sf("model"); { - std::ofstream ofs(WPATH(sf.filename()), OUTPUT_MODE); - CHECK(model_proto.SerializeToOstream(&ofs)); + auto output = filesystem::NewWritableFile(sf.filename()); + output->Write(model_proto.SerializeAsString()); } SentencePieceProcessor sp; @@ -1029,8 +1030,8 @@ TEST(SentencePieceProcessorTest, VocabularyTest) { test::ScopedTempFile sf("vocab.txt"); auto GetInlineFilename = [&sf](const std::string content) { { - io::OutputBuffer out(sf.filename()); - out.Write(content); + auto out = filesystem::NewWritableFile(sf.filename()); + out->Write(content); } return sf.filename(); }; |