6 #include <boost/test/unit_test.hpp> 8 #include "../QuantizationDataSet.hpp" 20 struct CsvTestHelper {
24 BOOST_TEST_MESSAGE(
"setup fixture");
29 BOOST_TEST_MESSAGE(
"teardown fixture");
33 std::string CreateTempCsvFile(std::map<
int, std::vector<float>> csvData)
35 fs::path fileDir = fs::temp_directory_path();
38 fs::path tensorInput1{fileDir /
"input_0_0.raw"};
39 fs::path tensorInput2{fileDir /
"input_1_0.raw"};
40 fs::path tensorInput3{fileDir /
"input_2_0.raw"};
46 std::ofstream ofs1{tensorInput1};
47 std::ofstream ofs2{tensorInput2};
48 std::ofstream ofs3{tensorInput3};
51 for(
auto entry : csvData.at(0))
55 for(
auto entry : csvData.at(1))
59 for(
auto entry : csvData.at(2))
64 ofs <<
"0, 0, " << tensorInput1.c_str() << std::endl;
65 ofs <<
"2, 0, " << tensorInput3.c_str() << std::endl;
66 ofs <<
"1, 0, " << tensorInput2.c_str() << std::endl;
73 catch (std::exception &e)
75 std::cerr <<
"Unable to write to file at location [" << p.c_str() <<
"] : " << e.what() << std::endl;
94 fs::remove(m_CsvFile.value());
96 catch (std::exception &e)
98 std::cerr <<
"Unable to delete file [" << m_CsvFile.value() <<
"] : " << e.what() << std::endl;
113 std::map<int, std::vector<float>> csvData;
114 csvData.insert(std::pair<
int, std::vector<float>>(0, { 0.111111f, 0.222222f, 0.333333f }));
115 csvData.insert(std::pair<
int, std::vector<float>>(1, { 0.444444f, 0.555555f, 0.666666f }));
116 csvData.insert(std::pair<
int, std::vector<float>>(2, { 0.777777f, 0.888888f, 0.999999f }));
118 std::string myCsvFile = CsvTestHelper::CreateTempCsvFile(csvData);
120 BOOST_TEST(!dataSet.
IsEmpty());
125 BOOST_TEST(input.GetPassId() == csvRow);
127 BOOST_TEST(input.GetLayerBindingIds().size() == 1);
128 BOOST_TEST(input.GetLayerBindingIds()[0] == 0);
129 BOOST_TEST(input.GetDataForEntry(0).size() == 3);
133 BOOST_TEST(input.GetDataForEntry(0).at(0) == csvData.at(csvRow).at(0));
134 BOOST_TEST(input.GetDataForEntry(0).at(1) == csvData.at(csvRow).at(1));
135 BOOST_TEST(input.GetDataForEntry(0).at(2) == csvData.at(csvRow).at(2));
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
QuantizationDataSet is a structure which is created after parsing a quantization CSV file...
BOOST_AUTO_TEST_SUITE_END()
fs::path NamedTempFile(const char *fileName)
Construct a temporary file name.
BOOST_FIXTURE_TEST_CASE(CheckDataSet, CsvTestHelper)