def test_formatting(self): server = MriServerDispatch({'id': 'abcde'}, HTTP_BIN, 'test', 'tester') event = TrainingEvent({'iteration': 100, 'loss': 200, 'accuracy': 300}, 'iteration') data = server._format_train_request(event) correct = json.dumps( {"type": "train.abcde", "properties": {"iteration": 100, "loss": 200, "accuracy": 300}} ) self.assertEqual(data, correct)
def test_format_train_request(self): server = MriServerDispatch({'title': 'test', 'id': 'cbdcig'}, HTTP_BIN, 'test', 'tester') event = TrainingEvent({'iteration': 100, 'loss': 0.5, 'accuracy': 0.6}, 'iteration') payload = server._format_train_request(event) self.assertTrue('train.cbdcig' in payload) self.assertTrue('"accuracy": 0.6' in payload)