python/jc
1
0
Fork 0
mirror of https://github.com/kellyjonbrazil/jc.git synced 2025-06-17 00:07:37 +02:00
Code Issues Releases Activity
Files
ee3b873ef46de2e447404d22db804304e51f9ab5
jc/tests/test_asciitable_m.py

396 lines
22 KiB
Python
Raw Normal View History

add tests
2022-03-21 17:57:14 -07:00
import os
import unittest
fix rstrip and add tests
2022-03-21 19:10:02 -07:00
from jc.exceptions import ParseError
add tests
2022-03-21 17:57:14 -07:00
import jc.parsers.asciitable_m
THIS_DIR = os.path.dirname(os.path.abspath(__file__))
class MyTests(unittest.TestCase):
def test_asciitable_m_nodata(self):
"""
Test 'asciitable_m' with no data
"""
self.assertEqual(jc.parsers.asciitable_m.parse('', quiet=True), [])
def test_asciitable_m_pure_ascii(self):
"""
Test 'asciitable_m' with a pure ASCII table
"""
input = '''
+========+========+========+========+========+========+========+
| type | tota | used | fr ee | shar | buff | avai |
| | l | | | ed | _cac | labl |
| | | | | | he | e |
+========+========+========+========+========+========+========+
| Mem | 3861 | 2228 | 3364 | 1183 | 2743 | 3389 |
| | 332 | 20 | 176 | 2 | 36 | 588 |
+--------+--------+--------+--------+--------+--------+--------+
| | | | | | | |
| | | | | test 2 | | |
+--------+--------+--------+--------+--------+--------+--------+
| last | last | last | ab cde | | | final |
+========+========+========+========+========+========+========+
'''
expected = [
{
"type": "Mem",
"tota_l": "3861\n332",
"used": "2228\n20",
"fr_ee": "3364\n176",
"shar_ed": "1183\n2",
"buff_cac_he": "2743\n36",
"avai_labl_e": "3389\n588"
},
{
fix tests so blank strings are now None
2022-03-22 12:25:51 -07:00
"type": None,
"tota_l": None,
"used": None,
"fr_ee": None,
add tests
2022-03-21 17:57:14 -07:00
"shar_ed": "test 2",
fix tests so blank strings are now None
2022-03-22 12:25:51 -07:00
"buff_cac_he": None,
"avai_labl_e": None
add tests
2022-03-21 17:57:14 -07:00
},
{
"type": "last",
"tota_l": "last",
"used": "last",
"fr_ee": "ab cde",
fix tests so blank strings are now None
2022-03-22 12:25:51 -07:00
"shar_ed": None,
"buff_cac_he": None,
add tests
2022-03-21 17:57:14 -07:00
"avai_labl_e": "final"
}
]
self.assertEqual(jc.parsers.asciitable_m.parse(input, quiet=True), expected)
def test_asciitable_m_unicode(self):
"""
Test 'asciitable_m' with a unicode table
"""
input = '''
╒════════╤════════╤════════╤════════╤════════╤════════╤════════╕
│ type │ tota │ used │ fr ee │ shar │ buff │ avai │
│ │ l │ │ │ ed │ _cac │ labl │
│ │ │ │ │ │ he │ e │
╞════════╪════════╪════════╪════════╪════════╪════════╪════════╡
│ Mem │ 3861 │ 2228 │ 3364 │ 1183 │ 2743 │ 3389 │
│ │ 332 │ 20 │ 176 │ 2 │ 36 │ 588 │
├────────┼────────┼────────┼────────┼────────┼────────┼────────┤
│ Swap │ 2097 │ 0 │ 2097 │ │ │ │
│ │ 148 │ │ 148 │ │ │ │
│ │ │ │ kb │ │ │ │
├────────┼────────┼────────┼────────┼────────┼────────┼────────┤
│ last │ last │ last │ ab cde │ │ │ final │
╘════════╧════════╧════════╧════════╧════════╧════════╧════════╛
'''
expected = [
{
"type": "Mem",
"tota_l": "3861\n332",
"used": "2228\n20",
"fr_ee": "3364\n176",
"shar_ed": "1183\n2",
"buff_cac_he": "2743\n36",
"avai_labl_e": "3389\n588"
},
{
"type": "Swap",
"tota_l": "2097\n148",
"used": "0",
"fr_ee": "2097\n148\nkb",
fix tests so blank strings are now None
2022-03-22 12:25:51 -07:00
"shar_ed": None,
"buff_cac_he": None,
"avai_labl_e": None
add tests
2022-03-21 17:57:14 -07:00
},
{
"type": "last",
"tota_l": "last",
"used": "last",
"fr_ee": "ab cde",
fix tests so blank strings are now None
2022-03-22 12:25:51 -07:00
"shar_ed": None,
"buff_cac_he": None,
add tests
2022-03-21 17:57:14 -07:00
"avai_labl_e": "final"
}
]
self.assertEqual(jc.parsers.asciitable_m.parse(input, quiet=True), expected)
def test_asciitable_m_pure_ascii_extra_spaces(self):
"""
Test 'asciitable_m' with a pure ASCII table that has heading and
trailing spaces and newlines.
"""
input = '''
+========+========+========+========+========+========+========+
| type | tota | used | fr ee | shar | buff | avai
| | l | | | ed | _cac | labl
| | | | | | he | e |
+========+========+========+========+========+========+========+
| Mem | 3861 | 2228 | 3364 | 1183 | 2743 | 3389 |
| | 332 | 20 | 176 | 2 | 36 | 588 |
+--------+--------+--------+--------+--------+--------+--------+
| | | | | | | |
| | | | | test 2 | | |
+--------+--------+--------+--------+--------+--------+--------+
| last | last | last | ab cde | | | final
+========+========+========+========+========+========+========+
'''
expected = [
{
"type": "Mem",
"tota_l": "3861\n332",
"used": "2228\n20",
"fr_ee": "3364\n176",
"shar_ed": "1183\n2",
"buff_cac_he": "2743\n36",
"avai_labl_e": "3389\n588"
},
{
fix tests so blank strings are now None
2022-03-22 12:25:51 -07:00
"type": None,
"tota_l": None,
"used": None,
"fr_ee": None,
add tests
2022-03-21 17:57:14 -07:00
"shar_ed": "test 2",
fix tests so blank strings are now None
2022-03-22 12:25:51 -07:00
"buff_cac_he": None,
"avai_labl_e": None
add tests
2022-03-21 17:57:14 -07:00
},
{
"type": "last",
"tota_l": "last",
"used": "last",
"fr_ee": "ab cde",
fix tests so blank strings are now None
2022-03-22 12:25:51 -07:00
"shar_ed": None,
"buff_cac_he": None,
add tests
2022-03-21 17:57:14 -07:00
"avai_labl_e": "final"
}
]
self.assertEqual(jc.parsers.asciitable_m.parse(input, quiet=True), expected)
def test_asciitable_m_unicode_extra_spaces(self):
"""
Test 'asciitable_m' with a pure ASCII table that has heading and
trailing spaces and newlines.
"""
input = '''
╒════════╤════════╤════════╤════════╤════════╤════════╤════════╕
type │ tota │ used │ free │ shar │ buff │ avai
│ l │ │ │ ed │ _cac │ labl
│ │ │ │ │ he │ e
╞════════╪════════╪════════╪════════╪════════╪════════╪════════╡
Mem │ 3861 │ 2228 │ 3364 │ 1183 │ 2743 │ 3389
│ 332 │ 20 │ 176 │ 2 │ 36 │ 588
├────────┼────────┼────────┼────────┼────────┼────────┼────────┤
Swap │ 2097 │ 0 │ 2097 │ │ │
│ 148 │ │ 148 │ │ │
╘════════╧════════╧════════╧════════╧════════╧════════╧════════╛
'''
expected = [
{
"type": "Mem",
"tota_l": "3861\n332",
"used": "2228\n20",
"free": "3364\n176",
"shar_ed": "1183\n2",
"buff_cac_he": "2743\n36",
"avai_labl_e": "3389\n588"
},
{
"type": "Swap",
"tota_l": "2097\n148",
"used": "0",
"free": "2097\n148",
fix tests so blank strings are now None
2022-03-22 12:25:51 -07:00
"shar_ed": None,
"buff_cac_he": None,
"avai_labl_e": None
add tests
2022-03-21 17:57:14 -07:00
}
]
self.assertEqual(jc.parsers.asciitable_m.parse(input, quiet=True), expected)
add bold bar seperator and ANSI code tests
2022-03-22 15:55:59 -07:00
def test_asciitable_m_pretty_ansi(self):
"""
Test 'asciitable-m' with a pretty table with ANSI codes
"""
input = '''
┏━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━┓
┃\x1b[1m \x1b[0m\x1b[1mReleased \x1b[0m\x1b[1m \x1b[0m┃\x1b[1m \x1b[0m\x1b[1mTitle \x1b[0m\x1b[1m \x1b[0m┃\x1b[1m \x1b[0m\x1b[1m Box Office\x1b[0m\x1b[1m \x1b[0m┃
┡━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━┩
│\x1b[36m \x1b[0m\x1b[36mDec 20, 2019\x1b[0m\x1b[36m \x1b[0m│\x1b[35m \x1b[0m\x1b[35mStar Wars: The Rise of Skywalker \x1b[0m\x1b[35m \x1b[0m│\x1b[32m \x1b[0m\x1b[32m $952,110,690\x1b[0m\x1b[32m \x1b[0m│
│\x1b[36m \x1b[0m\x1b[36mMay 25, 2018\x1b[0m\x1b[36m \x1b[0m│\x1b[35m \x1b[0m\x1b[35mSolo: A Star Wars Story \x1b[0m\x1b[35m \x1b[0m│\x1b[32m \x1b[0m\x1b[32m $393,151,347\x1b[0m\x1b[32m \x1b[0m│
│\x1b[36m \x1b[0m\x1b[36mDec 15, 2017\x1b[0m\x1b[36m \x1b[0m│\x1b[35m \x1b[0m\x1b[35mStar Wars Ep. V111: The Last Jedi\x1b[0m\x1b[35m \x1b[0m│\x1b[32m \x1b[0m\x1b[32m$1,332,539,889\x1b[0m\x1b[32m \x1b[0m│
│\x1b[36m \x1b[0m\x1b[36mDec 16, 2016\x1b[0m\x1b[36m \x1b[0m│\x1b[35m \x1b[0m\x1b[35mRogue One: A Star Wars Story \x1b[0m\x1b[35m \x1b[0m│\x1b[32m \x1b[0m\x1b[32m$1,332,439,889\x1b[0m\x1b[32m \x1b[0m│
└──────────────┴───────────────────────────────────┴────────────────┘
'''
expected = [
{
"released": "Dec 20, 2019\nMay 25, 2018\nDec 15, 2017\nDec 16, 2016",
"title": "Star Wars: The Rise of Skywalker\nSolo: A Star Wars Story\nStar Wars Ep. V111: The Last Jedi\nRogue One: A Star Wars Story",
"box_office": "$952,110,690\n$393,151,347\n$1,332,539,889\n$1,332,439,889"
}
]
self.assertEqual(jc.parsers.asciitable_m.parse(input, quiet=True), expected)
fix for special characters in headers
2022-03-23 15:08:33 -07:00
def test_asciitable_m_special_chars_in_header(self):
"""
Test 'asciitable_m' with a pure ASCII table that has special
characters in the header. These should be converted to underscores
and no trailing or consecutive underscores should end up in the
resulting key names.
"""
input = '''
+----------+------------+-----------+----------------+-------+--------------------+
| Protocol | Address | Age (min) | Hardware Addr | Type | Interface |
| | | of int | | | |
+----------+------------+-----------+----------------+-------+--------------------+
| Internet | 10.12.13.1 | 98 | 0950.5785.5cd1 | ARPA | FastEthernet2.13 |
+----------+------------+-----------+----------------+-------+--------------------+
'''
expected = [
{
"protocol": "Internet",
"address": "10.12.13.1",
"age_min_of_int": "98",
"hardware_addr": "0950.5785.5cd1",
"type": "ARPA",
"interface": "FastEthernet2.13"
}
]
self.assertEqual(jc.parsers.asciitable_m.parse(input, quiet=True), expected)
preserve keyname case with -r
2022-06-15 11:12:43 -07:00
def test_asciitable_no_lower_raw(self):
"""
Test 'asciitable_m' with a pure ASCII table that has special
characters and mixed case in the header. These should be converted to underscores
and no trailing or consecutive underscores should end up in the
resulting key names. Using `raw` in this test to preserve case. (no lower)
"""
input = '''
+----------+------------+-----------+----------------+-------+--------------------+
| Protocol | Address | Age (min) | Hardware Addr | Type | Interface |
| | | of int | | | |
+----------+------------+-----------+----------------+-------+--------------------+
| Internet | 10.12.13.1 | 98 | 0950.5785.5cd1 | ARPA | FastEthernet2.13 |
+----------+------------+-----------+----------------+-------+--------------------+
'''
expected = [
{
"Protocol": "Internet",
"Address": "10.12.13.1",
"Age_min_of_int": "98",
"Hardware_Addr": "0950.5785.5cd1",
"Type": "ARPA",
"Interface": "FastEthernet2.13"
}
]
self.assertEqual(jc.parsers.asciitable_m.parse(input, raw=True, quiet=True), expected)
add sep char test
2022-05-02 11:15:41 -07:00
def test_asciitable_m_sep_char_in_cell(self):
"""
Test 'asciitable_m' with a column separator character inside the data
"""
input = '''
| Author | yada | yada2 | yada3 | yada4 | yada5 | yada6 | yada7 |
├─────────────────┼─────────────┼─────────────────┼─────────────────┼─────────────────┼─────────────────┼────────────┼─────────────────┤
│ Kelly Brazil │ │ a76d46f9ecb1eff │ kellyjonbrazil@ │ Fri Feb 4 12:14 │ refactor ignore │ 1644005656 │ │
│ │ │ 4d6cc7ad633c97c │ gmail.com │ :16 2022 -0800 │ _exceptions │ │ │
│ │ │ ec0e99001a │ │ │ │ │ │
├─────────────────┼─────────────┼─────────────────┼─────────────────┼─────────────────┼─────────────────┼────────────┼─────────────────┤
│ Kevin Lyter │ │ 6b069a82d0fa19c │ lyterk@sent.com │ Thu Feb 3 18:13 │ Add xrandr to l │ 1643940838 │ │
│ │ │ 8d83b19b934bace │ │ :58 2022 -0800 │ ib.py │ │ │
│ │ │ 556cb758d7 │ │ │ │ │ │
├─────────────────┼─────────────┼─────────────────┼─────────────────┼─────────────────┼─────────────────┼────────────┼─────────────────┤
│ Kevin Lyter │ │ 6b793d052147406 │ lyterk@sent.com │ Thu Feb 3 18:13 │ Clean up types │ 1643940791 │ │
│ │ │ f388c4d5dc04f50 │ │ :11 2022 -0800 │ │ │ │
│ │ │ 6a3456f409 │ │ │ │ │ │
│ │ │ │ │ │ * | operator = │ │ │
│ │ │ │ │ │ > Union[] │ │ │
│ │ │ │ │ │ * Rem │ │ │
│ │ │ │ │ │ ove unused impo │ │ │
│ │ │ │ │ │ rt Iterator │ │ │
│ │ │ │ │ │ * R │ │ │
│ │ │ │ │ │ emove comment │ │ │
├─────────────────┼─────────────┼─────────────────┼─────────────────┼─────────────────┼─────────────────┼────────────┼─────────────────┤
│ Kevin Lyter │ │ ce9103f7cc66689 │ lyterk@sent.com │ Thu Feb 3 18:12 │ Delete old file │ 1643940766 │ │
│ │ │ 5dc7840d32797d8 │ │ :46 2022 -0800 │ s in template f │ │ │
│ │ │ c7274cf1b8 │ │ │ older │ │ │
├─────────────────┼─────────────┼─────────────────┼─────────────────┼─────────────────┼─────────────────┼────────────┼─────────────────┤
'''
expected = [
{
"author": "Kelly Brazil",
"yada": None,
"yada2": "a76d46f9ecb1eff\n4d6cc7ad633c97c\nec0e99001a",
"yada3": "kellyjonbrazil@\ngmail.com",
"yada4": "Fri Feb 4 12:14\n:16 2022 -0800",
"yada5": "refactor ignore\n_exceptions",
"yada6": "1644005656",
"yada7": None
},
{
"author": "Kevin Lyter",
"yada": None,
"yada2": "6b069a82d0fa19c\n8d83b19b934bace\n556cb758d7",
"yada3": "lyterk@sent.com",
"yada4": "Thu Feb 3 18:13\n:58 2022 -0800",
"yada5": "Add xrandr to l\nib.py",
"yada6": "1643940838",
"yada7": None
},
{
"author": "Kevin Lyter",
"yada": None,
"yada2": "ce9103f7cc66689\n5dc7840d32797d8\nc7274cf1b8",
"yada3": "lyterk@sent.com",
"yada4": "Thu Feb 3 18:12\n:46 2022 -0800",
"yada5": "Delete old file\ns in template f\nolder",
"yada6": "1643940766",
"yada7": None
}
]
optimize tests
2022-09-23 14:02:27 -07:00
self.assertEqual(jc.parsers.asciitable_m.parse(input, quiet=True), expected)
add sep char test
2022-05-02 11:15:41 -07:00
fix rstrip and add tests
2022-03-21 19:10:02 -07:00
def test_asciitable_m_markdown(self):
"""
Test 'asciitable_m' with a markdown table. Should raise a ParseError
"""
input = '''
| type | total | used | free | shared | buff cache | available |
|--------|---------|--------|---------|----------|--------------|-------------|
| Mem | 3861332 | 222820 | 3364176 | 11832 | 274336 | 3389588 |
| Swap | 2097148 | 0 | 2097148 | | | |
'''
self.assertRaises(ParseError, jc.parsers.asciitable_m.parse, input, quiet=True)
def test_asciitable_m_simple(self):
"""
Test 'asciitable_m' with a simple table. Should raise a ParseError
"""
input = '''
type total used free shared buff cache available
------ ------- ------ ------- -------- ------------ -----------
Mem 3861332 222820 3364176 11832 274336 3389588
Swap 2097148 0 2097148
'''
self.assertRaises(ParseError, jc.parsers.asciitable_m.parse, input, quiet=True)
add tests
2022-03-21 17:57:14 -07:00
if __name__ == '__main__':
unittest.main()
Reference in New Issue Copy Permalink