Fixed script for generating UT templates

Signed-off-by: Michal Mielewczyk <michal.mielewczyk@intel.com>
2019-01-21 11:18:48 -05:00 · 2019-01-21 11:18:48 -05:00 · 5e97795630
commit 5e97795630
parent 5c81824a8d
1 changed files with 3 additions and 0 deletions
--- a/tests/unit/framework/add_new_test_file.py
+++ b/tests/unit/framework/add_new_test_file.py
@ -48,6 +48,9 @@ class TestGenerator(object):
 		buf += self.get_empty_test_function()
 		buf += self.get_test_main()

+		with open(dst_path, "w") as f:
+			f.writelines(buf)
+
 		print dst_path + " generated successfully!"

 	def get_markups(self):