class TestModularServer(TestCase):
""" Test server for testing """
def portrayal(self, cell):
return {
"Shape": "rect",
"w": 1,
"h": 1,
"Filled": "true",
"Layer": 0,
"x": 0,
"y": 0,
"Color": "black"
}
def setUp(self):
self.user_params = {
'width': 1,
'height': 1,
'key1': UserSettableParameter('number', "Test Parameter", 101),
'key2': UserSettableParameter('slider', "Test Parameter", 200, 0, 300, 10)
}
self.viz_elements = [
CanvasGrid(self.portrayal, 10, 10, 20, 20),
TextElement(),
# ChartModule([{"Label": "Wolves", "Color": "#AA0000"}, # Todo - test chart module
# {"Label": "Sheep", "Color": "#666666"}])
]
self.server = ModularServer(MockModel, self.viz_elements, "Test Model", model_params=self.user_params)
def test_canvas_render_model_state(self):
test_portrayal = self.portrayal(None)
test_grid_state = defaultdict(list)
test_grid_state[test_portrayal['Layer']].append(test_portrayal)
state = self.server.render_model()
assert state[0] == test_grid_state
def test_text_render_model_state(self):
state = self.server.render_model()
assert state[1] == '<b>VisualizationElement goes here</b>.'
def test_user_params(self):
print(self.server.user_params)
assert self.server.user_params == {
'key1': UserSettableParameter('number', "Test Parameter", 101).json,
'key2': UserSettableParameter('slider', "Test Parameter", 200, 0, 300, 10).json
}
class TestModularServer(TestCase):
"""Test server for testing"""
def portrayal(self, cell):
return {
"Shape": "rect",
"w": 1,
"h": 1,
"Filled": "true",
"Layer": 0,
"x": 0,
"y": 0,
"Color": "black",
}
def setUp(self):
self.user_params = {
"width":
1,
"height":
1,
"key1":
UserSettableParameter("number", "Test Parameter", 101),
"key2":
UserSettableParameter("slider", "Test Parameter", 200, 0, 300, 10),
}
self.viz_elements = [
CanvasGrid(self.portrayal, 10, 10, 20, 20),
TextElement(),
# ChartModule([{"Label": "Wolves", "Color": "#AA0000"}, # Todo - test chart module
# {"Label": "Sheep", "Color": "#666666"}])
]
self.server = ModularServer(MockModel,
self.viz_elements,
"Test Model",
model_params=self.user_params)
def test_canvas_render_model_state(self):
test_portrayal = self.portrayal(None)
test_grid_state = defaultdict(list)
test_grid_state[test_portrayal["Layer"]].append(test_portrayal)
state = self.server.render_model()
assert state[0] == test_grid_state
def test_text_render_model_state(self):
state = self.server.render_model()
assert state[1] == "<b>VisualizationElement goes here</b>."
def test_user_params(self):
print(self.server.user_params)
assert self.server.user_params == {
"key1":
UserSettableParameter("number", "Test Parameter", 101).json,
"key2":
UserSettableParameter("slider", "Test Parameter", 200, 0, 300,
10).json,
}
