from __future__ import unicode_literals
import unittest, frappe, 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')