def test_add_children(self):
ref = Group()
child1 = Session()
child2 = Session()
ref.add_child(child1)
ref.add_child(child2)
self.assertEqual([child1, child2], ref.children)
def run(args):
"""Run the model with RMSE Loss.
It can train or test with different datasets.
"""
env.run_local = args.run_local
if not args.random_seed:
random.seed(args.seed)
tf.random.set_seed(args.seed)
if args.dry_run:
dry_run.run(args.enable_tf_caching, args.skip_non_cached)
model = create_model(args.model)
session = Session(
model=model, batch_size=args.batch_size, skip_non_cached=args.skip_non_cached,
)
if args.train:
optimizer = optimizers.Adam(args.lr)
session.train(
optimizer=optimizer,
cache_file=args.cache_file,
enable_checkpoint=not args.no_checkpoint,
epochs=args.epochs,
checkpoint=args.checkpoint,
)
if args.test is not None:
session.test(args.test)
def start_game(self, board_name):
Log.debug('Starting game')
#### TEMP ####
board_path = self.CORE.board_dir + board_name + self.CORE.board_suff
self.session = Session(self)
board = Board(self.session, board_path)
self.session.set_board(board)
self.player = Player('Local', 'temp_pwd')
self.session.add_player(self.player)
################
self.starting_game = True
def test_load_config(self):
# Crash if session has no name in config
ref = Session()
ref.load_configuration({
'path': 'test_path',
'anatomy_path': 'test_anatomy_path',
'name': 'test_name',
'description': 'test_desc',
'plot_settings': 'test_settings'
})
self.assertEqual(ref.name, 'test_name')
self.assertEqual(ref.description, 'test_desc')
self.assertEqual(ref.plot_settings, 'test_settings')
def test_get_configuration(self):
ref = Session()
ref.load_sequence('src/tests/test-data/brain.nii')
ref.load_anatomy('src/tests/test-data/mask.nii')
ref.load_stimuli('src/tests/test-data/stimuli.mat', 0.5)
expected = {
'path': 'src/tests/test-data/brain.nii',
'stimuli': {
'path': 'src/tests/test-data/stimuli.mat',
'tr': 0.5
},
'anatomy_path': 'src/tests/test-data/mask.nii'
}
self.assertEqual(ref.get_configuration(), expected)
def test_settings_changed(self):
ref = Session()
self.assertFalse(ref.settings_changed(False, False, None, None))
self.assertTrue(ref.settings_changed(True, False, None, None))
def test_load_anatomy(self, mock_brain):
ref = Session()
ref.load_sequence('src/tests/test-data/brain.nii')
mock_brain.assert_called_once_with('src/tests/test-data/brain.nii')
def test_session(self, mock_load):
Session('test.json')
mock_load.assert_called_once_with('test.json')
def add_session(self, name, detail, convener, attendee_capacity):
session = Session(name, detail, convener, attendee_capacity)
self._sessionList.append(session)
import time, hashlib, hmac
from flask import current_app
from src.session import Session
COINBASE_SESSION = Session()
class Client:
def __init__(self):
pass
def _request(self, method, url, data=None, extra_headers=None, **kwargs):
headers = {
'Accept': 'application/json',
'CB-ACCESS_KEY': current_app.config['API_KEY'],
'CB-ACCESS-SIGN': self.generate_cb_access_sign(method, url, data),
'CB-ACCESS-TIMESTAMP': str(int(time.time()))
}
if extra_headers:
headers.update(extra_headers)
return COINBASE_SESSION.request(method, url, **kwargs)
def get(self, url, data=None, headers=None, **kwargs):
return self._request('get', url, data, headers, **kwargs)
def generate_cb_access_sign(self, method, url, data=''):
message = f'{str(int(time.time()))}{method.upper()}{url}{data}'
signature = hmac.new(current_app.config['SECRET_KEY'].encode('utf-8'),
message.encode('utf-8'),
def setUp(self):
self.scaling_ghi = preprocessing.min_max_scaling_ghi()
self.session = Session(mock.Mock())