def testImportDataIntegers(self): test_filename = "test_file" test_file = open(test_filename, "w") test_file.write("time,value\n") testarray = [] for n in range(0, 100): testarray.append(1) for n in range(0, 20): if(n < 10): testarray.append(300) if(n >= 10): testarray.append(40) for n in range(0, 100): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + "," + str(1) + "\n") for n in range(0, 10): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + ",high" + "\n") for n in range(0, 10): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + ",low" + "\n") test_file.close() testobject = idata(test_filename) testallints = 0 for n in testobject._value: if isinstance(n, int): testallints = testallints + 1 self.assertEqual(120, testallints)
def testImportDataHighLow(self): test_filename = "test_file" test_file = open(test_filename, "w") test_file.write("time,value\n") testarray = [] for n in range(0, 100): testarray.append(1) for n in range(0, 20): if(n < 10): testarray.append(300) if(n >= 10): testarray.append(40) for n in range(0, 100): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + "," + str(1) + "\n") for n in range(0, 10): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + ",high" + "\n") for n in range(0, 10): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + ",low" + "\n") test_file.close() testobject = idata(test_filename) self.assertEqual(testarray, testobject._value[0: 120])
def testRoundTimeArraySum(self): test_filename = "activity_small.csv" test_file = open(test_filename, "w") test_file.write("time,value\n") for n in range(0, 100): tempdate = str(datetime.datetime(2019, 1, 1, 1, 12, 10)) test_file.write(tempdate + "," + str(1) + "\n") test_file.close() testobject = idata(test_filename) self.assertEqual([(datetime.datetime(2019, 1, 1, 1, 10, 10), 100)], rta(testobject, 5))
def testLinearSearchWrongInput(self): test_filename = "test_file" test_file = open(test_filename, "w") test_file.write("time,value\n") for n in range(0, 100): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + "," + str(1) + "\n") test_file.close() testobject = idata(test_filename) testobject.linear_search_value(datetime.datetime.now()) with self.assertRaises(Exception) as ex: testobject.linear_search_value(76) output = str(ex.exception) self.assertEqual(output, "key_time requires type datetime")
def testImportDataEmptyValues(self): test_filename = "test_file" test_file = open(test_filename, "w") test_file.write("time,value\n") for n in range(0, 100): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + "," + str(1) + "\n") for n in range(0, 10): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + "," + "\n") test_file.close() testobject = idata(test_filename) self.assertEqual(100, len(testobject._value))
def testTimeDateTime(self): test_filename = "test_file" test_file = open(test_filename, "w") test_file.write("time,value\n") for n in range(0, 100): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + "," + str(1) + "\n") for n in range(0, 10): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + ",high" + "\n") for n in range(0, 10): tempdate = str(datetime.datetime.now()) test_file.write(tempdate + ",low" + "\n") test_file.close() testobject = idata(test_filename) testalldatetimes = 0 for n in testobject._time: if isinstance(n, datetime.datetime): testalldatetimes = testalldatetimes + 1 self.assertEqual(120, testalldatetimes)