Commit 7b409f84 authored by Rémi Huguet's avatar Rémi Huguet
Browse files

feat: improve TraceHeaderSet and Trace representations and strings

parent 76bc347d

estraces/traces/trace.py

+4 −3
Original line number Diff line number Diff line
@@ -65,11 +65,12 @@ class Trace: 
        return self._metadatas



    def __repr__(self):

        return f'{self.__class__.__name__}(trace_id={self._id}, reader={self._reader})'

        return str(self)



    def __str__(self):

        r = 'Trace\n'

        r += f'{"Id":.<17}: {self._id}\n'

        r = 'Trace:\n'

        r += f'{"Reader instance":.<17}: {self._reader}\n'

        r += f'{"Index in set":.<17}: {self._id}\n'

        r += f'{"Samples size":.<17}: {len(self)}\n'

        for k in self.metadatas.keys():

            r += f'{k:.<17}: {self.metadatas.get(k)}\n'

estraces/traces/trace_header_set.py

+1 −3
Original line number Diff line number Diff line
@@ -130,9 +130,7 @@ class TraceHeaderSet: 
        return r



    def __repr__(self):

        return (

            f'{build_trace_header_set.__name__}(reader={self._reader}, name="{self.name}")'

        )

        return str(self)



    def __len__(self):

        return len(self.traces)

tests/test_trace_header_set.py

+5 −0
Original line number Diff line number Diff line
@@ -270,3 +270,8 @@ def test_split_raises_error_with_improper_type(ths): 
        split["jj"]

        split[-15]

        split[1:2]





def test_str_and_repr(ths):

    assert repr(ths) == str(ths)

    assert repr(ths[0]) == str(ths[0])