nanovna-saver/test/test_sitools.py

145 wiersze
6.2 KiB
Python

# NanoVNASaver
#
# A python program to view and export Touchstone data from a NanoVNA
# Copyright (C) 2019, 2020 Rune B. Broberg
# Copyright (C) 2020 NanoVNA-Saver Authors
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
import unittest
# Import targets to be tested
from NanoVNASaver.SITools import Format, Value
from decimal import Decimal
from math import inf
F_DEFAULT = Format()
F_ASSERT_DIGITS_0 = Format(max_nr_digits=0)
F_ASSERT_DIGITS_2 = Format(max_nr_digits=31)
F_ASSERT_OFFSET_1 = Format(min_offset=-9)
F_ASSERT_OFFSET_2 = Format(max_offset=9)
F_ASSERT_OFFSET_3 = Format(min_offset=9)
F_ASSERT_CLAMP = Format(parse_clamp_min=10, parse_clamp_max=9)
F_DIGITS_1 = Format(max_nr_digits=1, min_offset=-2,
max_offset=2, assume_infinity=False)
F_DIGITS_3 = Format(max_nr_digits=3, min_offset=-2,
max_offset=2, assume_infinity=False)
F_DIGITS_4 = Format(max_nr_digits=4)
F_DIGITS_31 = Format(max_nr_digits=31)
F_WITH_SPACE = Format(space_str=" ")
F_WITH_UNDERSCORE = Format(space_str="_")
class TestTSIToolsValue(unittest.TestCase):
def test_format_assertions(self):
self.assertRaises(AssertionError, Value, fmt=F_ASSERT_DIGITS_0)
self.assertRaises(AssertionError, Value, fmt=F_ASSERT_DIGITS_2)
self.assertRaises(AssertionError, Value, fmt=F_ASSERT_OFFSET_1)
self.assertRaises(AssertionError, Value, fmt=F_ASSERT_OFFSET_2)
self.assertRaises(AssertionError, Value, fmt=F_ASSERT_OFFSET_3)
self.assertRaises(AssertionError, Value, fmt=F_ASSERT_CLAMP)
def test_representation(self):
a = Value(1)
b = eval(repr(a))
self.assertEqual(repr(a), repr(b))
def test_default_format(self):
self.assertEqual(str(Value(0)), "0.00000")
self.assertEqual(str(Value(1)), "1.00000")
self.assertEqual(str(Value(10)), "10.0000")
self.assertEqual(str(Value(100)), "100.000")
self.assertEqual(str(Value(-1)), "-1.00000")
self.assertEqual(str(Value(-10)), "-10.0000")
self.assertEqual(str(Value(-100)), "-100.000")
self.assertEqual(str(Value(1e3)), "1.00000k")
self.assertEqual(str(Value(1e4)), "10.0000k")
self.assertEqual(str(Value(1e5)), "100.000k")
self.assertEqual(str(Value(1e6)), "1.00000M")
self.assertEqual(str(Value(1e7)), "10.0000M")
self.assertEqual(str(Value(1e8)), "100.000M")
self.assertEqual(str(Value(1e9)), "1.00000G")
self.assertEqual(str(Value(1e12)), "1.00000T")
self.assertEqual(str(Value(1e15)), "1.00000P")
self.assertEqual(str(Value(1e18)), "1.00000E")
self.assertEqual(str(Value(1e21)), "1.00000Z")
self.assertEqual(str(Value(1e24)), "1.00000Y")
self.assertEqual(str(Value(1e27)), "\N{INFINITY}")
self.assertEqual(str(Value(-1e27)), "-\N{INFINITY}")
self.assertEqual(float(Value(1e27)), 1e27)
self.assertEqual(str(Value(.1)), "100.000m")
self.assertEqual(str(Value(.01)), "10.0000m")
self.assertEqual(str(Value(.001)), "1.00000m")
self.assertEqual(str(Value(1e-6)), "1.00000µ")
self.assertEqual(str(Value(1e-9)), "1.00000n")
self.assertEqual(str(Value(1e-12)), "1.00000p")
self.assertEqual(str(Value(1e-15)), "1.00000f")
self.assertEqual(str(Value(1e-18)), "1.00000a")
self.assertEqual(str(Value(1e-21)), "1.00000z")
self.assertEqual(str(Value(1e-24)), "1.00000y")
self.assertEqual(str(Value(1e-27)), "0.00100y")
self.assertEqual(str(Value(1e-29)), "0.00001y")
self.assertEqual(str(Value(-1e-29)), "-0.00001y")
self.assertEqual(str(Value(1e-30)), "0.00000")
self.assertEqual(float(Value(1e-30)), 1e-30)
def test_format_digits_1(self):
v = Value(fmt=F_DIGITS_1)
self.assertEqual(str(v.parse("1")), "1")
self.assertEqual(str(v.parse("10")), "10")
self.assertEqual(str(v.parse("100")), "100")
self.assertEqual(str(v.parse("1e3")), "1k")
self.assertEqual(str(v.parse("1e4")), "10k")
self.assertEqual(str(v.parse("1e5")), "100k")
self.assertEqual(str(v.parse("1e9")), "1000M")
self.assertEqual(str(v.parse("1e-1")), "100m")
self.assertEqual(str(v.parse("1e-2")), "10m")
self.assertEqual(str(v.parse("1e-3")), "1m")
self.assertEqual(str(v.parse("1e-9")), "0")
def test_format_digits_3(self):
v = Value(fmt=F_DIGITS_3)
self.assertEqual(str(v.parse("1")), "1.00")
self.assertEqual(str(v.parse("10")), "10.0")
self.assertEqual(str(v.parse("100")), "100")
self.assertEqual(str(v.parse("1e3")), "1.00k")
self.assertEqual(str(v.parse("1e4")), "10.0k")
self.assertEqual(str(v.parse("1e5")), "100k")
self.assertEqual(str(v.parse("1e9")), "1000M")
self.assertEqual(str(v.parse("1e-1")), "100m")
self.assertEqual(str(v.parse("1e-2")), "10.0m")
self.assertEqual(str(v.parse("1e-3")), "1.00m")
self.assertEqual(str(v.parse("1e-9")), "0.00")
def test_format_digits_4(self):
v = Value(fmt=F_DIGITS_4)
self.assertEqual(str(v.parse("1")), "1.000")
self.assertEqual(str(v.parse("10")), "10.00")
self.assertEqual(str(v.parse("100")), "100.0")
self.assertEqual(str(v.parse("1e3")), "1.000k")
self.assertEqual(str(v.parse("1e4")), "10.00k")
self.assertEqual(str(v.parse("1e5")), "100.0k")
self.assertEqual(str(v.parse("1e-1")), "100.0m")
self.assertEqual(str(v.parse("1e-2")), "10.00m")
self.assertEqual(str(v.parse("1e-3")), "1.000m")
# TODO: test F_DIGITS_31
# F_WITH_SPACE
# F_WITH_UNDERSCORE