def test_idx_datapoint_exists(self): """Testing function idx_datapoint_exists.""" # Testing with known good value expected = True result = db_datapoint.idx_datapoint_exists( self.expected['idx_datapoint']) self.assertEqual(result, expected) # Testing with known bad value expected = False result = db_datapoint.idx_datapoint_exists(None) self.assertEqual(result, expected)
def _insert_datapoint(self): """Insert first datapoint in the database. Args: None Returns: None """ # Insert if db_datapoint.idx_datapoint_exists(1) is False: record = Datapoint(id_datapoint=general.encode(self.reserved), agent_label=general.encode(self.reserved), agent_source=general.encode(self.reserved)) database = db.Database() database.add(record, 1047)