style: format code with black
diff --git a/erpnext/tests/ui_test_helpers.py b/erpnext/tests/ui_test_helpers.py
index 9c8c371..44834c8 100644
--- a/erpnext/tests/ui_test_helpers.py
+++ b/erpnext/tests/ui_test_helpers.py
@@ -9,54 +9,67 @@
frappe.db.sql("DELETE FROM tabEmployee WHERE company='Test Org Chart'")
- emp1 = create_employee('Test Employee 1', 'CEO')
- emp2 = create_employee('Test Employee 2', 'CTO')
- emp3 = create_employee('Test Employee 3', 'Head of Marketing and Sales', emp1)
- emp4 = create_employee('Test Employee 4', 'Project Manager', emp2)
- emp5 = create_employee('Test Employee 5', 'Engineer', emp2)
- emp6 = create_employee('Test Employee 6', 'Analyst', emp3)
- emp7 = create_employee('Test Employee 7', 'Software Developer', emp4)
+ emp1 = create_employee("Test Employee 1", "CEO")
+ emp2 = create_employee("Test Employee 2", "CTO")
+ emp3 = create_employee("Test Employee 3", "Head of Marketing and Sales", emp1)
+ emp4 = create_employee("Test Employee 4", "Project Manager", emp2)
+ emp5 = create_employee("Test Employee 5", "Engineer", emp2)
+ emp6 = create_employee("Test Employee 6", "Analyst", emp3)
+ emp7 = create_employee("Test Employee 7", "Software Developer", emp4)
employees = [emp1, emp2, emp3, emp4, emp5, emp6, emp7]
return employees
+
@frappe.whitelist()
def get_employee_records():
- return frappe.db.get_list('Employee', filters={
- 'company': 'Test Org Chart'
- }, pluck='name', order_by='name')
+ return frappe.db.get_list(
+ "Employee", filters={"company": "Test Org Chart"}, pluck="name", order_by="name"
+ )
+
def create_company():
- company = frappe.db.exists('Company', 'Test Org Chart')
+ company = frappe.db.exists("Company", "Test Org Chart")
if not company:
- company = frappe.get_doc({
- 'doctype': 'Company',
- 'company_name': 'Test Org Chart',
- 'country': 'India',
- 'default_currency': 'INR'
- }).insert().name
+ company = (
+ frappe.get_doc(
+ {
+ "doctype": "Company",
+ "company_name": "Test Org Chart",
+ "country": "India",
+ "default_currency": "INR",
+ }
+ )
+ .insert()
+ .name
+ )
return company
+
def create_employee(first_name, designation, reports_to=None):
- employee = frappe.db.exists('Employee', {'first_name': first_name, 'designation': designation})
+ employee = frappe.db.exists("Employee", {"first_name": first_name, "designation": designation})
if not employee:
- employee = frappe.get_doc({
- 'doctype': 'Employee',
- 'first_name': first_name,
- 'company': 'Test Org Chart',
- 'gender': 'Female',
- 'date_of_birth': getdate('08-12-1998'),
- 'date_of_joining': getdate('01-01-2021'),
- 'designation': designation,
- 'reports_to': reports_to
- }).insert().name
+ employee = (
+ frappe.get_doc(
+ {
+ "doctype": "Employee",
+ "first_name": first_name,
+ "company": "Test Org Chart",
+ "gender": "Female",
+ "date_of_birth": getdate("08-12-1998"),
+ "date_of_joining": getdate("01-01-2021"),
+ "designation": designation,
+ "reports_to": reports_to,
+ }
+ )
+ .insert()
+ .name
+ )
return employee
+
def create_missing_designation():
- if not frappe.db.exists('Designation', 'CTO'):
- frappe.get_doc({
- 'doctype': 'Designation',
- 'designation_name': 'CTO'
- }).insert()
+ if not frappe.db.exists("Designation", "CTO"):
+ frappe.get_doc({"doctype": "Designation", "designation_name": "CTO"}).insert()