Code extension in the unit test for TSDB Wrapper API