brotherton-erpnext/erpnext/tests/test_regional.py

from __future__ import unicode_literals

import unittest

import frappe

import erpnext


@erpnext.allow_regional
def test_method():
	return 'original'

class TestInit(unittest.TestCase):
	def test_regional_overrides(self):
		frappe.flags.country = 'India'
		self.assertEqual(test_method(), 'overridden')

		frappe.flags.country = 'Maldives'
		self.assertEqual(test_method(), 'original')

		frappe.flags.country = 'France'
		self.assertEqual(test_method(), 'overridden')