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'])
