9 #include <boost/test/unit_test.hpp> 21 BOOST_TEST_MESSAGE(
"setup fixture");
26 BOOST_TEST_MESSAGE(
"teardown fixture");
30 std::string CreateTempCsvFile()
36 ofs <<
"airplane, bicycle , bird , \"m,o,n,k,e,y\"\n";
37 ofs <<
"banana, shoe, \"ice\"";
39 }
catch (std::exception &e)
41 std::cerr <<
"Unable to write to file at location [" << p.c_str() <<
"] : " << e.what() << std::endl;
49 int CheckStringsMatch(
CsvRow &row,
unsigned int index, std::string expectedValue)
51 return row.
values.at(index).compare(expectedValue);
65 fs::remove(m_CsvFile.value());
67 catch (std::exception &e)
69 std::cerr <<
"Unable to delete file [" << m_CsvFile.value() <<
"] : " << e.what() << std::endl;
84 std::vector<std::string> csvStrings;
85 csvStrings.reserve(2);
86 csvStrings.push_back(
"airplane, automobile , bird , \"c,a,t\"");
87 csvStrings.push_back(
"banana, shoe, \"ice\"");
89 std::vector<CsvRow> row = reader.
ParseVector(csvStrings);
93 BOOST_CHECK(row.size() == 2);
95 BOOST_CHECK(row1.
values.size() == 4);
96 BOOST_CHECK(CheckStringsMatch(row1, 0,
"airplane") == 0);
97 BOOST_CHECK(CheckStringsMatch(row1, 1,
"automobile") == 0);
98 BOOST_CHECK(CheckStringsMatch(row1, 2,
"bird") == 0);
99 BOOST_CHECK(CheckStringsMatch(row1, 3,
"c,a,t") == 0);
101 BOOST_CHECK(row2.
values.size() == 3);
102 BOOST_CHECK(CheckStringsMatch(row2, 0,
"banana") == 0);
103 BOOST_CHECK(CheckStringsMatch(row2, 1,
"shoe") == 0);
104 BOOST_CHECK(CheckStringsMatch(row2, 2,
"ice") == 0);
110 std::string theFilePath = TestHelper::CreateTempCsvFile();
112 std::vector<CsvRow> row = reader.
ParseFile(theFilePath);
116 BOOST_CHECK(row.size() == 2);
118 BOOST_CHECK(row1.
values.size() == 4);
119 BOOST_CHECK(CheckStringsMatch(row1, 0,
"airplane") == 0);
120 BOOST_CHECK(CheckStringsMatch(row1, 1,
"bicycle") == 0);
121 BOOST_CHECK(CheckStringsMatch(row1, 2,
"bird") == 0);
122 BOOST_CHECK(CheckStringsMatch(row1, 3,
"m,o,n,k,e,y") == 0);
124 BOOST_CHECK(row2.
values.size() == 3);
125 BOOST_CHECK(CheckStringsMatch(row2, 0,
"banana") == 0);
126 BOOST_CHECK(CheckStringsMatch(row2, 1,
"shoe") == 0);
127 BOOST_CHECK(CheckStringsMatch(row2, 2,
"ice") == 0);
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(TestParseVector, TestHelper)
std::vector< std::string > values
static std::vector< CsvRow > ParseVector(const std::vector< std::string > &csvVector)
BOOST_AUTO_TEST_SUITE_END()
fs::path NamedTempFile(const char *fileName)
Construct a temporary file name.
static std::vector< CsvRow > ParseFile(const std::string &csvFile)