def test_writer_headers():
args = main.get_runtime_params()
args.outFilename = "tests/out/sample.csv"
fileWriter = writer.Writer(args, main.outputHeaders)
fileWriter.write_headers()
with open(args.outFilename) as csvFile:
headers = csvFile.readline().strip()
assert headers == main.outputHeaders
os.remove(args.outFilename)
def test_reader_batch_average():
args = main.get_runtime_params()
args.inFilename = "tests/data/testDataSmall.csv"
args.averageResults = True
fileReader = reader.Reader(args)
headers = fileReader.get_headers()
expectedResult = [
{headers[0]: fileReader.convert_string_to_time('2020-03-27 04.00.00'),
headers[1]: (1.662758 + 1.710038) / 2,
headers[2]: (-3.345694 - 3.842012) / 2}]
acutalResult = [row for row in fileReader.get_one_batch()]
assert expectedResult == acutalResult
def test_reader_batch_one_line():
args = main.get_runtime_params()
args.inFilename = "tests/data/testDataSmall.csv"
fileReader = reader.Reader(args)
headers = fileReader.get_headers()
expectedResult = []
expectedResult.append({headers[0]: fileReader.convert_string_to_time('2020-03-27 04.00.00'),
headers[1]: 1.662758,
headers[2]: -3.345694})
expectedResult.append({headers[0]: fileReader.convert_string_to_time('2020-03-27 04.01.00'),
headers[1]: 1.710038,
headers[2]: -3.842012})
acutalResult = [row for row in fileReader.get_one_batch()]
assert expectedResult == acutalResult
def test_writer_one_line():
args = main.get_runtime_params()
args.outFilename = "tests/out/sample.csv"
fileWriter = writer.Writer(args, main.outputHeaders)
time = datetime.strptime("2020-10-01 10:10:09", '%Y-%m-%d %H:%M:%S')
uComp = -2.231231
vComp = 1.231231
WindData = converter.WindRecord(time, uComp, vComp)
fileWriter.write_one_line(WindData)
with open(args.outFilename) as csvFile:
resultTime, resultSpeed, resultDegree = csvFile.readline().strip(
).split(",")
assert datetime.strptime(resultTime,
fileWriter.get_time_format()) == time
assert resultSpeed == "%.6f" % WindData.get_wind_speed()
assert resultDegree == "%.6f" % WindData.get_wind_direction_deg()
os.remove(args.outFilename)
def test_writer_init_creates_file():
args = main.get_runtime_params()
args.outFilename = "tests/out/sample.csv"
writer.Writer(args, main.outputHeaders)
assert os.path.isfile(args.outFilename)
os.remove(args.outFilename)
def test_reader_init_reads_header():
args = main.get_runtime_params()
args.inFilename = "tests/data/testDataSmall.csv"
fileReader = reader.Reader(args)
assert fileReader.get_headers() == 'datetime,u-comp,v-comp'.split(",")