Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
adriank committed Nov 24, 2018
1 parent 11d877b commit 2c4d2b4
Showing 1 changed file with 67 additions and 22 deletions.
89 changes: 67 additions & 22 deletions tests/test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,27 +8,72 @@
sys.setrecursionlimit(20000)

class Utils_test(unittest.TestCase):
def test_Utils_JSON_compat(self):
self.assertEqual(loads("['ddd']"),['ddd'])
if sys.version_info.major < 3:
self.assertEqual(loads("[u'ddd']"),['ddd'])
self.assertRaises(Exception, lambda: loads(['ddd}']))
self.assertEqual(dumps(['ddd']),'["ddd"]')
self.assertEqual(py2JSON(False), 'false')
self.assertEqual(py2JSON(None), 'null')
self.assertEqual(py2JSON((2,3,4)), [2,3,4])
if sys.version_info.major < 3:
self.assertEqual(py2JSON(unicode('')), '')
self.assertEqual(py2JSON(2), 2)
self.assertEqual(printJSON([1,2,3,4,5,6]), "[\n \x1b[36m\x1b[1m1\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m2\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m3\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m4\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m5\x1b[0m\x1b[0m,\n ... (1 more items)\n]")
self.assertEqual(printJSON([{},1]), '[\n {},\n \x1b[36m\x1b[1m1\x1b[0m\x1b[0m\n]')
self.assertEqual(printJSON({"aaa":1}), '{\x1b[33m\x1b[1m"aaa"\x1b[0m\x1b[0m: \x1b[36m\x1b[1m1\x1b[0m\x1b[0m}')
self.assertEqual(printJSON({"a":[1,2,3]}), '{\x1b[33m\x1b[1m"a"\x1b[0m\x1b[0m: [\n \x1b[36m\x1b[1m1\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m2\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m3\x1b[0m\x1b[0m\n]}')
self.assertEqual(printJSON([[1],{"aa":2}]), '[\n [\x1b[36m\x1b[1m1\x1b[0m\x1b[0m],\n {\x1b[33m\x1b[1m"aa"\x1b[0m\x1b[0m: \x1b[36m\x1b[1m2\x1b[0m\x1b[0m}\n]')
self.assertEqual(printJSON({"aaa":{"bbb":{"ccc":{"ddd":[1,2,3,4,5]}}}}), '{\x1b[33m\x1b[1m"aaa"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"bbb"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"ccc"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"ddd"\x1b[0m\x1b[0m: [\n \x1b[36m\x1b[1m1\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m2\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m3\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m4\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m5\x1b[0m\x1b[0m\n]}}}}')
if str(sys.version_info.major)+str(sys.version_info.minor) < '33':
self.assertEqual(printJSON({"aaa":{"bbb":{"ccc":{"ddd":{"eee":[1,2,3,4,5],"ddd":{}}}}}}), '{\x1b[33m\x1b[1m"aaa"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"bbb"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"ccc"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"ddd"\x1b[0m\x1b[0m: {\n \x1b[33m\x1b[1m"eee"\x1b[0m\x1b[0m: <array of 5 items>,\n \x1b[33m\x1b[1m"ddd"\x1b[0m\x1b[0m: {...}\n}}}}}')
def test_Utils_JSON_compat(self):
self.assertEqual(loads("['ddd']"), ['ddd'])
if sys.version_info.major < 3:
self.assertEqual(loads("[u'ddd']"), ['ddd'])
self.assertRaises(Exception, lambda: loads(['ddd}']))
self.assertEqual(dumps(['ddd']), '["ddd"]')
self.assertEqual(py2JSON(False), 'false')
self.assertEqual(py2JSON(None), 'null')
self.assertEqual(py2JSON((2, 3, 4)), [2, 3, 4])
if sys.version_info.major < 3:
self.assertEqual(py2JSON(unicode('')), '')
self.assertEqual(py2JSON(2), 2)
self.assertEqual(
printJSON([1, 2, 3, 4, 5, 6]),
"[\n \x1b[36m\x1b[1m1\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m2\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m3\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m4\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m5\x1b[0m\x1b[0m,\n ... (1 more items)\n]"
)
self.assertEqual(
printJSON([{}, 1]), '[\n {},\n \x1b[36m\x1b[1m1\x1b[0m\x1b[0m\n]'
)
self.assertEqual(
printJSON({
"aaa": 1
}),
'{\x1b[33m\x1b[1m"aaa"\x1b[0m\x1b[0m: \x1b[36m\x1b[1m1\x1b[0m\x1b[0m}'
)
self.assertEqual(
printJSON({
"a": [1, 2, 3]
}),
'{\x1b[33m\x1b[1m"a"\x1b[0m\x1b[0m: [\n \x1b[36m\x1b[1m1\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m2\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m3\x1b[0m\x1b[0m\n]}'
)
self.assertEqual(
printJSON([[1], {
"aa": 2
}]),
'[\n [\x1b[36m\x1b[1m1\x1b[0m\x1b[0m],\n {\x1b[33m\x1b[1m"aa"\x1b[0m\x1b[0m: \x1b[36m\x1b[1m2\x1b[0m\x1b[0m}\n]'
)
self.assertEqual(
printJSON({
"aaa": {
"bbb": {
"ccc": {
"ddd": [1, 2, 3, 4, 5]
}
}
}
}),
'{\x1b[33m\x1b[1m"aaa"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"bbb"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"ccc"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"ddd"\x1b[0m\x1b[0m: [\n \x1b[36m\x1b[1m1\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m2\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m3\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m4\x1b[0m\x1b[0m,\n \x1b[36m\x1b[1m5\x1b[0m\x1b[0m\n]}}}}'
)
if str(sys.version_info.major) + str(sys.version_info.minor) < '33':
self.assertEqual(
printJSON({
"aaa": {
"bbb": {
"ccc": {
"ddd": {
"eee": [1, 2, 3, 4, 5],
"ddd": {}
}
}
}
}
}),
'{\x1b[33m\x1b[1m"aaa"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"bbb"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"ccc"\x1b[0m\x1b[0m: {\x1b[33m\x1b[1m"ddd"\x1b[0m\x1b[0m: {\n \x1b[33m\x1b[1m"eee"\x1b[0m\x1b[0m: <array of 5 items>,\n \x1b[33m\x1b[1m"ddd"\x1b[0m\x1b[0m: {...}\n}}}}}'
)

testcase1=unittest.TestLoader().loadTestsFromTestCase(Utils_test)
testcase1 = unittest.TestLoader().loadTestsFromTestCase(Utils_test)

utils_test=unittest.TestSuite([testcase1])
utils_test = unittest.TestSuite([testcase1])

0 comments on commit 2c4d2b4

Please sign in to comment.