# Path: vasppy/atom.py
# class Atom:
#     """
#     Class for individual atoms
#     """
# 
#     def __init__( self, label, r ): # currently assume fractional coordinates
#         """
#         Initialise an Atom instance
#     
#         Args:
#             label (Str): a label for this atom
#             r (numpy.array): the atom coordinates
#         
#         Returns:
#             None
#         """
#         self.label = label
#         self.r = r


import unittest
import numpy as np
from vasppy.atom import Atom


class AtomTestCase( unittest.TestCase ):

    def test_init_atom( self ):
        label = 'A'
        r = np.array( [ 0.1, 0.2, 0.3 ] )

# Path: vasppy/atom.py
# class Atom:
#     """
#     Class for individual atoms
#     """
# 
#     def __init__( self, label, r ): # currently assume fractional coordinates
#         """
#         Initialise an Atom instance
#     
#         Args:
#             label (Str): a label for this atom
#             r (numpy.array): the atom coordinates
#         
#         Returns:
#             None
#         """
#         self.label = label
#         self.r = r


# Path: tests/test_atom.py
import unittest
import numpy as np
from vasppy.atom import Atom



class AtomTestCase( unittest.TestCase ):

    def test_init_atom( self ):
        label = 'A'
        r = np.array( [ 0.1, 0.2, 0.3 ] )

        atom = Atom( label=label, r=r )