Beispiel #1
0
    def test_templated_input_arrays(self):

        template = "Anchor\n" + \
                   "0 0 0 0 0\n"

        outfile = open(self.templatename, 'w')
        outfile.write(template)
        outfile.close()

        gen = InputFileGenerator()
        gen.set_template_file(self.templatename)
        gen.set_generated_file(self.filename)

        gen.mark_anchor('Anchor')
        gen.transfer_array(array([1, 2, 3, 4.75, 5.0]), 1, 3, 5, sep=' ')

        gen.generate()

        infile = open(self.filename, 'r')
        result = infile.read()
        infile.close()

        answer = "Anchor\n" + \
                   "0 0 1.0 2.0 3.0 4.75 5.0\n"

        self.assertEqual(answer, result)