def testGeneratingCo2(self):
simul = ReeferSimulator()
df = simul.generateCo2Records(cid="C01",
nb_records=100,
product_id="P02")
self.assertIsNotNone(df)
self.assertEqual(df.size, 1800) # nb of rows x nbr of columns
append = True
elif arg == "--db":
useDB = True
return (cid, simulation_type, nb_records, product_id, fname, append, useDB)
if __name__ == "__main__":
(cid, simulation_type, nb_records, product_id, fname, append,
useDB) = parseArguments()
if useDB:
print(cid, simulation_type, nb_records, product_id)
else:
print(cid, simulation_type, nb_records, product_id, fname, append)
simulator = ReeferSimulator()
if simulation_type == ReeferSimulator.SIMUL_POWEROFF:
df = simulator.generatePowerOffRecords(cid, nb_records, product_id)
elif simulation_type == ReeferSimulator.SIMUL_CO2:
df = simulator.generateCo2Records(cid, nb_records, product_id)
elif simulation_type == ReeferSimulator.SIMUL_O2:
df = simulator.generateO2Records(cid, nb_records, product_id)
elif simulation_type == ReeferSimulator.SIMUL_TEMPERATURE:
df = simulator.generateTemperatureRecords(cid, nb_records, product_id)
elif simulation_type == ReeferSimulator.NORMAL:
df = simulator.generateNormalRecords(cid, nb_records, product_id)
else:
print("Not a valid simulation")
exit
if (not useDB):
saveFile(df, fname, append)
else:
saveToDB(df)