1414// limitations under the License.
1515// *****************************************************************************
1616
17+ #include < memory>
18+
1719#include < gtest/gtest.h>
1820
1921#include " ngraph/cpio.hpp"
@@ -40,25 +42,25 @@ TEST(cpio, read)
4042
4143 {
4244 int index = 0 ;
43- char * data = static_cast <char *>( malloc ( file_info[index].get_size ()) );
44- reader.read (file_info[index].get_name (), data, file_info[index].get_size ());
45- string content = string (data, file_info[index].get_size ());
45+ auto data = unique_ptr <char >( new char [ file_info[index].get_size ()] );
46+ reader.read (file_info[index].get_name (), data. get () , file_info[index].get_size ());
47+ string content = string (data. get () , file_info[index].get_size ());
4648 EXPECT_STREQ (content.c_str (), " 12345" );
4749 }
4850
4951 {
5052 int index = 1 ;
51- char * data = static_cast <char *>( malloc ( file_info[index].get_size ()) );
52- reader.read (file_info[index].get_name (), data, file_info[index].get_size ());
53- string content = string (data, file_info[index].get_size ());
53+ auto data = unique_ptr <char >( new char [ file_info[index].get_size ()] );
54+ reader.read (file_info[index].get_name (), data. get () , file_info[index].get_size ());
55+ string content = string (data. get () , file_info[index].get_size ());
5456 EXPECT_STREQ (content.c_str (), " this is a test" );
5557 }
5658
5759 {
5860 int index = 2 ;
59- char * data = static_cast <char *>( malloc ( file_info[index].get_size ()) );
60- reader.read (file_info[index].get_name (), data, file_info[index].get_size ());
61- string content = string (data, file_info[index].get_size ());
61+ auto data = unique_ptr <char >( new char [ file_info[index].get_size ()] );
62+ reader.read (file_info[index].get_name (), data. get () , file_info[index].get_size ());
63+ string content = string (data. get () , file_info[index].get_size ());
6264 EXPECT_STREQ (content.c_str (), " the quick brown fox jumped over the lazy dog" );
6365 }
6466}
@@ -90,17 +92,17 @@ TEST(cpio, write)
9092
9193 {
9294 int index = 0 ;
93- char * data = static_cast <char *>( malloc ( file_info[index].get_size ()) );
94- reader.read (file_info[index].get_name (), data, file_info[index].get_size ());
95- string content = string (data, file_info[index].get_size ());
95+ auto data = unique_ptr <char >( new char [ file_info[index].get_size ()] );
96+ reader.read (file_info[index].get_name (), data. get () , file_info[index].get_size ());
97+ string content = string (data. get () , file_info[index].get_size ());
9698 EXPECT_STREQ (content.c_str (), s1.c_str ());
9799 }
98100
99101 {
100102 int index = 1 ;
101- char * data = static_cast <char *>( malloc ( file_info[index].get_size ()) );
102- reader.read (file_info[index].get_name (), data, file_info[index].get_size ());
103- string content = string (data, file_info[index].get_size ());
103+ auto data = unique_ptr <char >( new char [ file_info[index].get_size ()] );
104+ reader.read (file_info[index].get_name (), data. get () , file_info[index].get_size ());
105+ string content = string (data. get () , file_info[index].get_size ());
104106 EXPECT_STREQ (content.c_str (), s2.c_str ());
105107 }
106108 }
0 commit comments