/
test_dataclasses.py
46 lines (33 loc) · 1.26 KB
/
test_dataclasses.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
"""
Test the scalarstop.dataclasses module.
"""
import dataclasses
import unittest
import scalarstop as sp
def _serialize_dataclass():
"""Serialize a dataclass."""
@sp.dataclasses.dataclass
class blob:
"""Our example dataclass."""
a: int
b: int
b = blob(1, 2)
return sp.pickle.dumps(b)
def _load_serialized_fields(s):
"""Deserialize a dataclass."""
return sp.pickle.loads(s)
class TestDataclass(unittest.TestCase):
"""Test our custom dataclass code."""
def setUp(self):
"""Serialize and deserialize a dataclass."""
self.dc = _load_serialized_fields(_serialize_dataclass())
def test_dataclasses_library__is_broken(self):
"""Test that the dataclasses library doesn't work on cloudpickle'd dataclasses."""
self.assertEqual(dataclasses.fields(self.dc), ())
self.assertEqual(dataclasses.astuple(self.dc), ())
self.assertEqual(dataclasses.asdict(self.dc), {})
def test_our_code_works(self):
"""Test that our versions of dataclasses code does work."""
self.assertEqual(len(sp.dataclasses.fields(self.dc)), 2)
self.assertEqual(sp.dataclasses.astuple(self.dc), (1, 2))
self.assertEqual(sp.dataclasses.asdict(self.dc), dict(a=1, b=2))