3699777599

Build Type

push

github

Committed by GitHub

Commit Message

Add initial mypy integration (#241)

Run Details

223 of 223 new or added lines in 45 files covered. (100.0%)

4575 of 6107 relevant lines covered (74.91%)

0.75 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

57.14

/winterdrp/data/source_data.py

from typing import Optional

import pandas as pd

from winterdrp.data.base_data import DataBatch, DataBlock


class SourceTable(DataBlock):
    def __init__(self, source_list: pd.DataFrame, metadata: dict):
        self.source_list = source_list
        self.metadata = metadata
        super().__init__()

    def get_data(self) -> pd.DataFrame:
        return self.source_list

    def set_data(self, source_list: pd.DataFrame):
        self.source_list = source_list

    def get_metadata(self) -> dict:
        return self.metadata

    def __getitem__(self, item):
        return self.metadata.__getitem__(item)

    def __setitem__(self, key, value):
        self.metadata.__setitem__(key, value)

    def keys(self):
        return self.metadata.keys()


class SourceBatch(DataBatch):

    data_type = SourceTable

    def __init__(self, batch: Optional[list[SourceTable] | SourceTable] = None):
        super().__init__(batch=batch)

    def append(self, data: SourceTable):
        self._append(data)

    def get_batch(self) -> list[SourceTable]:
        return self.get_data_list()

1	from typing import Optional	1✔
2
3	import pandas as pd	1✔
4
5	from winterdrp.data.base_data import DataBatch, DataBlock	1✔
6
7
8	class SourceTable(DataBlock):	1✔
9	def __init__(self, source_list: pd.DataFrame, metadata: dict):	1✔
10	self.source_list = source_list	×
11	self.metadata = metadata	×
12	super().__init__()	×
13
14	def get_data(self) -> pd.DataFrame:	1✔
15	return self.source_list	×
16
17	def set_data(self, source_list: pd.DataFrame):	1✔
18	self.source_list = source_list	×
19
20	def get_metadata(self) -> dict:	1✔
21	return self.metadata	×
22
23	def __getitem__(self, item):	1✔
24	return self.metadata.__getitem__(item)	×
25
26	def __setitem__(self, key, value):	1✔
27	self.metadata.__setitem__(key, value)	×
28
29	def keys(self):	1✔
30	return self.metadata.keys()	×
31
32
33	class SourceBatch(DataBatch):	1✔
34
35	data_type = SourceTable	1✔
36
37	def __init__(self, batch: Optional[list[SourceTable] \| SourceTable] = None):	1✔
38	super().__init__(batch=batch)	×
39
40	def append(self, data: SourceTable):	1✔
41	self._append(data)	×
42
43	def get_batch(self) -> list[SourceTable]:	1✔
44	return self.get_data_list()	×

winter-telescope / winterdrp / 3699777599

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous