mirror of
https://github.com/aljazceru/bitfinex-api-py.git
synced 2026-01-06 23:44:21 +01:00
Add bfxapi/tests/test_labeler.py unit test.
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
import unittest
|
||||
from .test_rest_serializers_and_types import TestRestSerializersAndTypes
|
||||
from .test_websocket_serializers_and_types import TestWebsocketSerializersAndTypes
|
||||
from .test_labeler import TestLabeler
|
||||
|
||||
NAME = "tests"
|
||||
|
||||
@@ -8,6 +9,7 @@ def suite():
|
||||
return unittest.TestSuite([
|
||||
unittest.makeSuite(TestRestSerializersAndTypes),
|
||||
unittest.makeSuite(TestWebsocketSerializersAndTypes),
|
||||
unittest.makeSuite(TestLabeler),
|
||||
])
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
24
bfxapi/tests/test_labeler.py
Normal file
24
bfxapi/tests/test_labeler.py
Normal file
@@ -0,0 +1,24 @@
|
||||
import unittest
|
||||
|
||||
from dataclasses import dataclass
|
||||
from ..exceptions import LabelerSerializerException
|
||||
from ..labeler import _Type, generate_labeler_serializer, generate_recursive_serializer
|
||||
|
||||
class TestLabeler(unittest.TestCase):
|
||||
def test_generate_labeler_serializer(self):
|
||||
@dataclass
|
||||
class Test(_Type):
|
||||
A: int
|
||||
B: float
|
||||
C: str
|
||||
|
||||
labels = [ "A", "_PLACEHOLDER", "B", "_PLACEHOLDER", "C" ]
|
||||
|
||||
serializer = generate_labeler_serializer("Test", Test, labels)
|
||||
|
||||
self.assertEqual(serializer.parse(5, None, 65.0, None, "X"), Test(5, 65.0, "X"))
|
||||
self.assertRaises(LabelerSerializerException, serializer.parse, 5, 65.0, "X")
|
||||
self.assertListEqual(serializer.get_labels(), [ "A", "B", "C" ])
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user