brotherton-erpnext/erpnext/tests/test_init.py

import unittest

import frappe

from erpnext import encode_company_abbr

test_records = frappe.get_test_records('Company')

class TestInit(unittest.TestCase):
	def test_encode_company_abbr(self):
		company = frappe.new_doc("Company")
		company.company_name = "New from Existing Company For Test"
		company.abbr = "NFECT"
		company.default_currency = "INR"
		company.save()

		abbr = company.abbr

		names = [
			"Warehouse Name", "ERPNext Foundation India", "Gold - Member - {a}".format(a=abbr),
			" - {a}".format(a=abbr), "ERPNext - Foundation - India",
			"ERPNext Foundation India - {a}".format(a=abbr),
			"No-Space-{a}".format(a=abbr), "- Warehouse"
		]

		expected_names = [
			"Warehouse Name - {a}".format(a=abbr), "ERPNext Foundation India - {a}".format(a=abbr),
			"Gold - Member - {a}".format(a=abbr), " - {a}".format(a=abbr),
			"ERPNext - Foundation - India - {a}".format(a=abbr),
			"ERPNext Foundation India - {a}".format(a=abbr), "No-Space-{a} - {a}".format(a=abbr),
			"- Warehouse - {a}".format(a=abbr)
		]

		for i in range(len(names)):
			enc_name = encode_company_abbr(names[i], company.name)
			self.assertTrue(
				enc_name == expected_names[i],
				"{enc} is not same as {exp}".format(enc=enc_name, exp=expected_names[i])
			)