source: nappy/trunk/nappy/unit_tests/test_na_file_1001.py @ 3349

"""
test_na_file_1001.py
====================

Tests for the na_file_1001.py module.

"""

# Import standard library modules
import unittest
import os
import sys

# Import local modules
os.environ["NAPPY_BASE_DIR"] = "..\\.."
base_dir = os.environ["NAPPY_BASE_DIR"]
sys.path.append(base_dir)

import nappy
import nappy.na_file.na_file

class NAFile1001_TestCase(unittest.TestCase):

    def setUp(self):
        self.infile = os.path.join(base_dir, "data_files", "1001.na")
        self.outfile = os.path.join(base_dir, "test_outputs", "test_1001.na")
        self.fin = nappy.openNAFile(self.infile)
        self.fin.readData()
        self.na_dict = self.fin.getNADict()        

    def test_read1001(self):
        "Tests reading FFI 1001."
        self.assertEqual(type(self.na_dict), type({1:2}))

    def test_write1001(self):
        "Tests writing FFI 1001."
        self.fobj = nappy.openNAFile(self.outfile, mode="w", na_dict=self.na_dict)              
        self.failUnless(isinstance(self.fobj, nappy.na_file.na_file.NAFile))

if __name__ ==  "__main__":

    unittest.main()