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(",")