def test_set_criteria(self):
a = Graph()
data = {
0: {
"1D": "A231",
"Gender": "F",
"Age": "23",
"Sales": "245",
"BMI": "Normal",
"Salary": "20 000",
"Birthday": "24/06/1994"
},
1: {
"ID": "B324",
"Gender": "M",
"Age": "25",
"Sales": "270",
"BMI": "Normal",
"Salary": "20 000",
"Birthday": "24/06/1992"
},
2: {
"ID": "C322",
"Gender": "M",
"Age": "26",
"Sales": "270",
"BMI": "Normal",
"Salary": "20 000",
"Birthday": "24/06/1991"
},
3: {
"ID": "D032",
"Gender": "M",
"Age": "27",
"Sales": "270",
"BMI": "Normal",
"Salary": "20 000",
"Birthday": "24/06/1990"
}
}
a.set_data(data, 'bar', 'data2.txt')
a.set_criteria("Gender", 'Female')
# a.set_data({"dfd": "asdfds"}, "bar", "C:\\temp\\random.html")
# a.graph_data = {0: {"ID": "A23", "Gender": "Male", "Age": 22, "Sales": 245, "BMI": "normal", "salary": 20,
# "Birthday": "24/06/1995"}}
a.graph_type.set_data_keys("Gender", "Sales")
expected = "F"
actual = a.graph_type.data[0]['Gender']
self.assertEqual(expected, actual)
class Controller:
def __init__(self):
self.db_handler = DatabaseHandler()
self.data = None
self.filehandler = None
self.graph = None
def load(self, filename):
"""
Set the file that will create the filehandler object
"""
if path.exists(filename):
self.filehandler = FileHandler(filename)
self.filehandler.set_file_type()
return True
else:
return False
def validate(self):
"""
Read selected file
"""
# James' changes (13/03)
result = self.filehandler.read()
self.data = result
print("")
print(result)
def set_local(self, connection):
"""
Set the local database with a specified name
:param connection:
:return:
"""
self.db_handler.set_local(connection)
self.db_handler.insert_local_dict(self.data)
def set_remote(self, host, user, password, db):
"""
Set the remote database
:param host, user, password, db:
:return:
"""
self.db_handler.set_remote(host, user, password, db)
self.db_handler.insert_remote_dict(self.data)
def set_graph(self, graph_type, filename):
print(graph_type)
print(filename)
self.graph = Graph()
data = self.data
self.graph.set_data(data, graph_type, filename)
def get_local(self):
self.data = self.db_handler.get_local()
def get_remote(self):
self.data = self.db_handler.get_remote()
def set_criteria(self, criteria_1, criteria_2=None):
self.graph.set_criteria(criteria_1, criteria_2)
def set_keys(self, key_1, key_2=None):
self.graph.set_keys(key_1, key_2)
def draw(self, x, y, title):
self.graph.draw(x, y, title)
def check_data(self):
if self.data is not None:
return True
return False
class TestChart(TestCase):
def setUp(self):
self.graph = Graph()
self.testArray = {
0: {
"ID": "A233",
"Gender": "M",
"Age": "22",
"Sales": "245",
"BMI": "Normal",
"Salary": "20",
"Birthday": "24/06/1995"
},
1: {
"ID": "A244",
"Gender": "F",
"Age": "30",
"Sales": "666",
"BMI": "Underweight",
"Salary": "301",
"Birthday": "05/05/1988"
},
2: {
"ID": "A525",
"Gender": "M",
"Age": "35",
"Sales": "456",
"BMI": "Obesity",
"Salary": "84",
"Birthday": "01/08/1983"
},
3: {
"ID": "A266",
"Gender": "M",
"Age": "24",
"Sales": "999",
"BMI": "Normal",
"Salary": "999",
"Birthday": "24/05/1993"
}
}
def tearDown(self):
self.graph = None
def test_chart_scatter(self):
self.graph.set_data(self.testArray, "scatter",
"expected_test\\scatter")
self.graph.set_criteria("Gender", "M")
self.graph.set_keys("Salary", "Sales")
result = self.graph.draw("Salary", "Sales", "Title")
def test_chart_pie(self):
self.graph.set_data(self.testArray, "pie", "expected_test\\pie")
self.graph.set_criteria("Gender", "M")
self.graph.set_keys("Salary", "Sales")
result = self.graph.draw("Salary", "Sales", "Title")
def test_chart_bar(self):
self.graph.set_data(self.testArray, "bar", "expected_test\\bar")
self.graph.set_criteria("Gender", "M")
self.graph.set_keys("Salary", "Sales")
result = self.graph.draw("Salary", "Sales", "Title")