Skip to content

Commit

Permalink
Create placeholder class for NormalizedDataManager
Browse files Browse the repository at this point in the history
  • Loading branch information
kravitsjacob committed Jan 8, 2025
1 parent 4d9400e commit c16a87b
Show file tree
Hide file tree
Showing 5 changed files with 55 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/paxplot/__init__.py
Original file line number Diff line number Diff line change
@@ -1,2 +1,4 @@
"""Paxplot modules"""
from .core import pax_parallel, PaxFigure
from .datasets import *
from .data_managers import *
2 changes: 2 additions & 0 deletions src/paxplot/data_managers/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
"""Data managers for paxplot"""
from .normalized import *
21 changes: 21 additions & 0 deletions src/paxplot/data_managers/normalized.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
"""Class for NormalizedDataManager"""


class NormalizedDataManager():
"""
Manages data and a normalized (between 0 and 1) representation of that same data
"""

def __init__(self):
self.true_data = []
self.normalized_data = []


def append(self, data=list):
"""Append data to manager. Updates both the true and normalized data.
Args:
data (_type_, optional): _description_. Defaults to list.
"""
self.true_data = data
self.normalized_data = data
Empty file.
30 changes: 30 additions & 0 deletions testing/data_managers/test_normalized.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
"""Tests for NormalizedDataManager"""

import unittest
import paxplot


class NormalizedDataManagerTests(unittest.TestCase):
"""Tests for NormalizedDataManager
Args:
unittest (_type_): _description_
"""

def test_append_success(self):
"""
Basic appending data
"""
# Arrange
data = [
[0.0, 0.0],
[1.0, 1.0],
[2.0, 2.0]
]
paxdataset = paxplot.data_managers.NormalizedDataManager()

# Act
paxdataset.append(data)

# Assert
self.assertEqual(paxdataset.true_data[0][0], 0.0)

0 comments on commit c16a87b

Please sign in to comment.