def test_encode_csv_vector_no_gen(): num_bins = 30 num_beams = 4 vel = EarthVelocity(num_bins, num_beams) # Populate data val = 2.0 for beam in range(vel.element_multiplier): for bin_num in range(vel.num_elements): vel.Velocities[bin_num][beam] = val vel.Magnitude, vel.Direction = EarthVelocity.generate_vectors( vel.Velocities) vel.remove_vessel_speed(-1.3, -1.4, 0.3) dt = datetime.datetime.now() # Create CSV lines result = vel.encode_csv(dt, 'A', 1, 1.3, 1.0) # Check the csv data for line in result: if bool(re.search(Ensemble.CSV_MAG, line[0])): assert bool(re.search(str(2.477), line[0])) assert bool(re.search(Ensemble.CSV_MAG, line[0])) elif bool(re.search(Ensemble.CSV_DIR, line[0])): assert bool(re.search(str(49.3987), line[0])) assert bool(re.search(Ensemble.CSV_DIR, line[0])) elif bool(re.search(Ensemble.CSV_EARTH_VEL, line[0])): assert True else: assert False
def test_encode_csv(): num_bins = 30 num_beams = 4 vel = EarthVelocity(num_bins, num_beams) # Populate data val = 1.0 for beam in range(vel.element_multiplier): for bin_num in range(vel.num_elements): vel.Velocities[bin_num][beam] = val val += 1.1 vel.Magnitude, vel.Direction = EarthVelocity.generate_vectors( vel.Velocities) dt = datetime.datetime.now() # Create CSV lines result = vel.encode_csv(dt, 'A', 1, 1.3, 1.0) # Check the csv data test_value = 1.0 for line in result: if bool(re.search(Ensemble.CSV_EARTH_VEL, line[0])): assert bool(re.search(str(test_value), line[0])) assert bool(re.search(Ensemble.CSV_EARTH_VEL, line[0])) test_value += 1.1