def __init__(self):
Cmd.__init__(self) # Initialize cmd interface here
self.prompt = ">>> " # Initialize prompt
self._vld = DataValidator(
) # Object of DataValidator, for validating data
self._std = EmployeeData() # Instance of EmployeeData
self.intro = "WELCOME TO THE EMPLOYEE DATABASE MANAGEMENT CONSOLE \n ENTER A KEYWORD TO START. FOT HELP, ENTER \"help\"."
def add_data(data):
"""
>>> add_data(["T123", "M", "28", "100", "Obesity", "350", "11-09-1990"]) #doctest: +NORMALIZE_WHITESPACE
[{'EMPID': 'T123', 'GENDER': 'M', 'AGE': '28', 'SALES': '100', 'BMI': 'Obesity',
'SALARY': '350', 'BIRTHDAY': '11-09-1990'}]
"""
sd = EmployeeData()
sd.add_data(data)
return sd.new_data
class EmployeeDataUnitTests(unittest.TestCase):
"""
http://istqbexamcertification.com/what-is-decision-coverage-its-advantages-and-disadvantages/#more-417
"""
def setUp(self):
self.employee = EmployeeData()
# print("A test case is called")
# ****************************************************************
# ************************************************************************
# employee_data
def tearDown(self):
print("This test case is done!")
def test_employee_info_no_header(self):
employees_data_no_header = [[
'A123', 'M', '25', '123', 'Normal', '39', '31-01-1992'
], ['T123', 'M', '20', '654', 'Normal', '56', '16/10/1996']]
result = self.employee.employee_info(employees_data_no_header)
# remove the header
self.assertEqual(employees_list_no_header, result[0])
def test_employee_info_with_header(self):
employees_data_with_header = [
['empid', 'gender', 'age', 'sales', 'bmi', 'salary', 'dob'],
['A123', 'M', '25', '123', 'Normal', '39', '31-01-1992'],
['T123', 'M', '20', '654', 'Normal', '56', '16/10/1996']
]
result = self.employee.data(employees_data_with_header)
# remove the header
employees_list_with_header.pop(0)
self.assertEqual(employees_data_with_header, result[0])
self.assertEqual([], result[1])
def test_employee_info_same_empid(self):
employees_data_duplicate_id = [[
'A123', 'M', '25', '123', 'Normal', '39', '31-01-1992'
], ['A123', 'M', '20', '654', 'Normal', '56', '16/10/1996']]
result = self.employee.employee_info(employees_data_duplicate_id)
# remove duplicate record
employees_data_duplicate_id.pop(1)
self.assertEqual(employees_data_duplicate_id, result[0])
self.assertEqual([['Record 2', 'EmpID not unique']], result[1])
def test_employee_info_invalid_record(self):
employees_data_invalid_record = [[
'A123', 'M', '25', '123', 'Normal', '39', '31-01-1992'
], ['T123', 'X', '20', '654', 'Normal', '56', '16/10/1996']]
result = self.employee.employee_info(employees_data_invalid_record)
# remove invalid record
employees_data_invalid_record.pop(1)
self.assertEqual(employees_data_invalid_record, result[0])
# error list
self.assertEqual([['Record 2', 'Gender : Not M or F']], result[1])
def test_18(self):
employee_data = EmployeeData()
employee_data.data = [{
"EMPID": "T123",
"AGE": 33,
"GENDER": "M",
"SALES": 200,
"BMI": "Normal",
"SALARY": 200,
"BIRTHDAY": "31-12-1985"}]
new_data = ["T123", 26, "F", 150, "Normal", 100, "01-01-1992"]
self.assertTrue(employee_data.data_exist(new_data))
def get_bmi():
"""
>>> get_bmi()
{'Obesity': 1, 'Normal': 2}
"""
sd = EmployeeData()
v = DataValidator()
sd.add_data(
v.check_all(["T133", "M", "28", "100", "Obesity", "350",
"11-09-1990"]))
sd.add_data(
v.check_all(["T134", "M", "29", "150", "Normal", "300", "12-08-1989"]))
sd.add_data(
v.check_all(["T135", "F", "30", "100", "Normal", "300", "13-07-1988"]))
return sd.get_bmi()
def setUp(self):
self.employee = EmployeeData()
def test_17(self):
employee_data = EmployeeData()
self.assertRaises(OSError, employee_data.save_data)
def test_16(self):
employee_data = EmployeeData()
employee_data.select_source("source")
self.assertIsNone(employee_data._source)
def test_15(self):
employee_data = EmployeeData()
employee_data.select_source("csv", "employeeinfo.csv")
self.assertIsInstance(employee_data._source, FileHandler)
class InterpreterController(Cmd):
# This class is a controller class which enable interpreter (model) and
# View (Console View) to interconnect with each other and
# This class also defined command loop enables \
# the associate commands invoke
#
# Written By: Patel
#
def __init__(self):
Cmd.__init__(self) # Initialize cmd interface here
self.prompt = ">>> " # Initialize prompt
self._vld = DataValidator(
) # Object of DataValidator, for validating data
self._std = EmployeeData() # Instance of EmployeeData
self.intro = "WELCOME TO THE EMPLOYEE DATABASE MANAGEMENT CONSOLE \n ENTER A KEYWORD TO START. FOT HELP, ENTER \"help\"."
# Welcome information
def do_select(self, line):
"""
Select a data source
:param line: <String> Source name
:return: None
:Author: Vaishali Patel
"""
# Available data sources
options = "-csv", "-db"
args = list(arg.lower() for arg in str(line).split())
try:
# Check if the input data source is available in this program or not
if args[0] not in options:
raise ValueError("The data resource is not available.")
else:
# Code for initialise CSV data source
if args[0] == "-csv":
try:
if len(args) == 1:
self._std.select_source(args[0][1:],
"employeeinfo.csv")
View.warning(
"No CSV file path specified. A default file \"employeeinfo.csv\" will be used."
)
elif len(args) == 2:
self._std.select_source(args[0][1:], args[1])
elif len(args) == 3:
if args[1] == "-a":
self._std.select_source(
args[0][1:], args[2], True)
except (CSVError, OSError) as e:
View.error(e)
except Exception as e:
View.error(e)
else:
View.success("Data source CSV is selected.")
# Code for initialise database source
elif args[0] == "-db":
try:
self._std.select_source(args[0][1:])
except (ConnectionError, TypeError) as e:
View.error(e)
except Exception as e:
View.error(e)
else:
View.success("Data source Database is selected.")
# Code for initialise XXXX data source
else:
pass
# Catch and display error message
except ValueError as e:
View.error(str(e) + "\n")
View.help_select()
except Exception as e:
View.error(e)
def do_add(self, line):
"""
# This function enable to add employee data into system
# Which basically give all the available option to the user
# parse: splits the given sequence of characters or values (text)
# into smaller parts based on some rules
#
# Written By: Patel
:param line: <EMPID> <Age> <Gender> <Sales> <BMI> <Salary> <Birthday>
:return: None
"""
# Split the input argument to obtain the data
raw_data = list(arg.lower() for arg in str(line).split())
try:
# Check if input data has 7 data fields
if not len(raw_data) == len(Data):
raise AttributeError("Please input correct data.")
else:
# Check and wash data by check_all() of DataValidator
result = self._vld.check_all(raw_data)
# Check if there is any None which stands for invalid input
if None in result:
key = 0
# build a list of name list
items = list(map(lambda i: i.name, Data))
e_str = ""
while key < len(result):
if result[key] is None:
# Left alignment
e_str += "{:<10}".format(items[key])
key += 1
raise ValueError("The following field(s) is invalid:\n%s" %
e_str)
else:
self._std.add_data(result)
except (AttributeError, ValueError) as e:
View.error(str(e) + "\n")
View.help_add()
except CSVError as e:
View.error(e)
except Exception as e:
View.error(e)
else:
View.success("Add data")
def do_save(self, arg):
"""
Save data to specified data source# This function enable to access dict option for respective file
# to be stored in the system
# Raise Exception error if unable to do so
#
# Written By: Patel
#
:param arg: arg
:return: None
"""
# If no data source selected, prompt user to do so.
try:
self._std.save_data()
except ValueError as e:
View.info(e)
except (OSError, AttributeError) as e:
View.error(e)
except Exception as e:
View.error(e)
else:
View.success("Data is saved")
def do_show(self, line):
# Get all instructions# This function enable to create dictionary option
# to respective commands and the system find matches
# to the respective option to show individual map / view
#
# Written By: Patel
#
args = list(arg.lower() for arg in str(line).split())
# Those commands are required single arguments
# single_commands = ["-a"]
# Those commands are required two arguments
plot_commands = ["-p", "-b", "-c"]
# Show data table
if args[0] == "-t":
if len(self._std.data) == 0 and len(self._std.new_data) == 0:
View.info("No data to display.")
if not len(self._std.data) == 0:
View.display("ORIGINAL DATA:")
View.display_data(self._std.data, ind=True)
if not len(self._std.new_data) == 0:
View.display("\nNEW DATA:")
View.display_data(self._std.new_data, ind=True)
View.display("\n(Input command \"save\" to save the new data)")
elif args[0] in plot_commands:
try:
if len(args) == 1:
raise IndexError("Incomplete command line.")
if args[0] == "-p":
self.show_pie(args[1])
if args[0] == "-b":
self.show_bar(args[1])
if args[0] == "-c":
self.show_scatter(args[1])
except IndexError as e:
View.error(str(e) + "\n")
View.help_show()
else:
View.info("Invalid command line.\n")
View.help_show()
def show_pie(self, line):
# Draw Pies
try:
if len(self._std.get_gender()) == 0 or len(
self._std.get_bmi()) == 0:
raise ValueError("No data to display.")
# Draw gender
if line.upper() == Data.GENDER.name:
View.plot_pie(self._std.get_gender(), "Gender Distribution")
# Draw BMI
if line.upper() == Data.BMI.name:
View.plot_pie(self._std.get_bmi(), "Body Mass Index (BMI)")
except ValueError as e:
View.info(e)
except Exception as e:
View.error(e)
def show_bar(self, line):
# Draw Bars
try:
if len(self._std.get_gender()) == 0 or len(
self._std.get_gender()) == 0:
raise ValueError("No data to display.")
# Draw gender
if line.upper() == Data.GENDER.name:
View.plot_bar(self._std.get_gender(), "Gender Distribution")
# Draw BMI
if line.upper() == Data.BMI.name:
View.plot_bar(self._std.get_bmi(), "Body Mass Index (BMI)")
except ValueError as e:
View.info(e)
except Exception as e:
View.error(e)
# Author: Vaishali Patel
def show_scatter(self, line):
# Draw Line Scatter
try:
if len(self._std.get_gender()) == 0 or len(
self._std.get_gender()) == 0:
raise ValueError("No data to display.")
# Draw gender
if line.upper() == Data.GENDER.name:
View.plot_bar(self._std.get_gender(), "Gender Distribution")
# Draw BMI
if line.upper() == Data.SALARY.name:
View.plot_bar(self._std.get_salary(),
"Body Mass Index (SALARY)")
except ValueError as e:
View.info(e)
except Exception as e:
View.error(e)
@staticmethod
def help_show():
View.help_show()
@staticmethod
def help_add():
View.help_add()
@staticmethod
def help_save():
View.help_save()
@staticmethod
def help_select():
View.help_select()
@staticmethod
def help_quit():
View.help_quit()
def do_quit(self, line):
arg = str(line).lower()
if not arg == "-f" and not len(self._std.new_data) == 0:
View.warning(
"The new data hasn't been saved. Enter \"quit -f\" to quit without saving."
)
else:
View.display("Thanks for using. Bye!")
return True
def setUp(self):
self.ctrl = InterpreterController(EmployeeData())
