def test_channel_messages_9(): data = get_data() owner_token, owner_id, u_token, u_id, channel_id, channel_id2 = user_setup( ) now = datetime.datetime.now() one_hour_later = (now + datetime.timedelta(hours=1)).timestamp() m_1 = message_send(owner_token, channel_id, "message1") m_2 = message_send(owner_token, channel_id, "message2") m_3 = message_send(owner_token, channel_id, "message3") m_4 = message_send(owner_token, channel_id, "message4") m_5 = message_sendlater(owner_token, channel_id, "message5 (future)", one_hour_later) m_6 = message_send(owner_token, channel_id, "message6") messagelist = data['messages'] messagelist.sort(key=lambda i: i['time_created'], reverse=True) ########################## END SETUP ######################## list1 = [] start = 0 end = 6 timestamp = datetime.datetime.now().timestamp() for i in range(start, end): if timestamp > messagelist[i]['time_created']: list1.append(messagelist[i]) assert (channel_messages(owner_token, channel_id, 0) == { 'messages': list1, 'start': 0, 'end': -1 })
def test_message_sendlater_20(): owner_token, owner_id, u_token, u_id = setup() now = datetime.datetime.now() one_hour_later = (now + datetime.timedelta(hours=1)).timestamp() channel = channels_create(owner_token, "Channel Name", 'true') channel_id = channel['channel_id'] channel_join(u_token, channel_id) assert message_sendlater(u_token, channel_id, "Future Message One", one_hour_later) == { 'message_id': 1 } assert message_sendlater(u_token, channel_id, "Future Message Two", one_hour_later) == { 'message_id': 2 } assert get_data()['messages'] == [{ 'message_id': 1, 'u_id': u_id, 'message': "Future Message One", 'time_created': one_hour_later, 'is_unread': True, 'reacts': [], 'is_pinned': False, }, { 'message_id': 2, 'u_id': u_id, 'message': "Future Message Two", 'time_created': one_hour_later, 'is_unread': True, 'reacts': [], 'is_pinned': False, }]
def test_message_edit_4(): data = get_data() user1, user_id1, admin1, admin_id1, admin2, admin_id2, channel1, channel2 = setup( ) message_send(user1, channel1, "hey admin, did you hear you can edit other people messages") message_send(admin1, channel1, 'yep, let me show you my admin rights') assert message_edit(admin1, 1, 'imagine not being an admin') == {} for messagedict in data['messages']: if messagedict['message_id'] == 1: assert messagedict['message'] == 'imagine not being an admin'
def test_message_edit_3(): data = get_data() user1, user_id1, admin1, admin_id1, admin2, admin_id2, channel1, channel2 = setup( ) message_send(admin1, channel1, "hey admin 2, apparently we can edit each others messages") message_send(admin2, channel1, 'that sounds pretty standard to me') assert message_edit(admin1, 2, 'admin1 is really cool') == {} for messagedict in data['messages']: if messagedict['message_id'] == 2: assert messagedict['message'] == 'admin1 is really cool'
def message_setup(): data = get_data() owner_token, owner_id, u_token, u_id, channel_id, channel_id2 = user_setup( ) # Loop through and add 52 messages to the message list for i in range(52): message_text = "message" + str(i) message = m_1 = message_send(owner_token, channel_id, message_text) messagelist = data['messages'] messagelist.sort(key=lambda i: i['time_created'], reverse=True) return messagelist
def setup(): reset_data() data = get_data() user_dict1 = auth_register('*****@*****.**', 'hello123', 'Steven', 'Lay') user1 = user_dict1['token'] user_id1 = user_dict1['u_id'] user_dict2 = auth_register('*****@*****.**', 'hello123', 'Steven2', 'Lay') user2 = user_dict2['token'] user_id2 = user_dict2['u_id'] channel_dict1 = channels_create(user1, 'chat1', 'true') channel1 = channel_dict1['channel_id'] channel_join(user2, channel1) return data, user1, user_id1, user2, user_id2, channel1
def test_message_sendlater_5(): owner_token, owner_id, u_token, u_id = setup() future_time = datetime.datetime(3000, 1, 1).timestamp() channel = channels_create(owner_token, "Channel Name", 'true') channel_id = channel['channel_id'] channel_join(u_token, channel_id) assert message_sendlater(u_token, channel_id, 1000 * "a", future_time) == { 'message_id': 1 } assert get_data()['messages'] == [{ 'message_id': 1, 'u_id': u_id, 'message': 1000 * "a", 'time_created': future_time, 'is_unread': True, 'reacts': [], 'is_pinned': False, }]
def test_message_sendlater_19(): owner_token, owner_id, u_token, u_id = setup() now = datetime.datetime.now() one_hour_later = (now + datetime.timedelta(hours=1)).timestamp() two_hours_later = (now + datetime.timedelta(hours=2)).timestamp() channel = channels_create(owner_token, "Channel Name", 'true') channel_id = channel['channel_id'] channel_join(u_token, channel_id) assert message_sendlater(u_token, channel_id, "Future Message One", one_hour_later) == { 'message_id': 1 } immediate_message_1 = message_send(u_token, channel_id, "Immediate Message One") assert message_sendlater(u_token, channel_id, "Future Message Two", two_hours_later) == { 'message_id': 3 } immediate_message_2 = message_send(u_token, channel_id, "Immediate Message Two") assert get_data()['messages'] == [ { 'message_id': 1, 'u_id': u_id, 'message': "Future Message One", 'time_created': one_hour_later, 'is_unread': True, 'reacts': [], 'is_pinned': False, }, message_dict(2), # from immediate message { 'message_id': 3, 'u_id': u_id, 'message': "Future Message Two", 'time_created': two_hours_later, 'is_unread': True, 'reacts': [], 'is_pinned': False, }, message_dict(4), # from immediate message ]
def setup(): reset_data() data = get_data() user_dict1 = auth_register('*****@*****.**', 'hello123', 'Steven', 'Lay') user1 = user_dict1['token'] user_id1 = user_dict1['u_id'] user = user_dict(user_id1) user['permission_id'] = 3 admin_dict1 = auth_register('*****@*****.**', 'adminhello123', 'adminSteven', 'Lay') admin1 = admin_dict1['token'] admin_id1 = admin_dict1['u_id'] admin = user_dict(admin_id1) admin['permission_id'] = 1 admin_dict2 = auth_register('*****@*****.**', 'adminhello123', 'adminSteven', 'Lay') admin2 = admin_dict2['token'] admin_id2 = admin_dict2['u_id'] admin_2 = user_dict(admin_id2) admin_2['permission_id'] = 1 admin_dict3 = auth_register('*****@*****.**', 'adminhello123', 'admin2Steven', 'Lay') admin3 = admin_dict3['token'] admin_id3 = admin_dict3['u_id'] admin_3 = user_dict(admin_id3) admin_3['permission_id'] = 1 channel_dict1 = channels_create(admin1, 'chat1', 'true') channel1 = channel_dict1['channel_id'] channel_dict2 = channels_create(user1, 'chat2', 'true') channel2 = channel_dict2['channel_id'] channel_join(user1, channel1) channel_join(admin2, channel1) return (data, user1, user_id1, admin1, admin_id1, admin2, admin_id2, admin3, admin_id3, channel1, channel2)
def setup(): reset_data() data = get_data() user_dict1 = auth_register('*****@*****.**', 'hello123', 'Steven', 'Lay') user1 = user_dict1['token'] user_id1 = user_dict1['u_id'] user_dict2 = auth_register('*****@*****.**', 'hello123', '2Steven', 'Lay') user2 = user_dict2['token'] user_id2 = user_dict2['u_id'] channel_dict1 = channels_create(user1, 'chat1', 'true') channel1 = channel_dict1['channel_id'] channel_dict2 = channels_create(user1, 'chat2', 'true') channel2 = channel_dict2['channel_id'] channel_join(user2, channel1) messagelist = data['messages'] return user1, user_id1, user2, user_id2, channel1, channel2, messagelist
args.pca) + str(args.features_out) hash_ = hashlib.sha1(params_to_hash.encode()).hexdigest() # Get the name of the ROI name_ROI = ['IFGorb', 'IFGtri', 'TP', 'TPJ', 'aSTS', 'pSTS', 'All'][args.ROI] # Generate the data and save it to a pickle print('Subject : ', args.subject, '| ROI : ', name_ROI, ' | File : ', args.file, ' | Hash : ', hash_) print('Generating data ...') dt.generate_data_subject(args.subject, args.ROI, name_ROI, args.pca, args.features_out, hash_, args.n_components) # Load the data from the pickle print('Loading data ...') data = dt.get_data(args.subject, args.ROI, name_ROI, hash_, args.shuffle) # Check outisde of the loop if our model-saving directory exists if not os.path.exists( os.path.join(paths.path2Output, 'models_blocks', 'Sub_{}'.format( args.subject), '{}'.format(name_ROI))): os.makedirs( os.path.join(paths.path2Output, 'models_blocks', 'Sub_{}'.format(args.subject), '{}'.format(name_ROI))) # Train the model print('Training ...') r2_tests, r2_trains = [], [] # Loop over blocks for block, current_block in tqdm(enumerate(data),
def floor_all_message_times(): for message in get_data()['messages']: floor_message_time(message['message_id'])