def test_shot_data(self): """Check that filenames with shot numbers are are used when requested.""" filename = os.path.join(os.path.dirname(__file__), "test_dsv_shot_(shot).dat") test_acq = DSVAcquisition() test_shot_number = 12345 test_data = test_acq.getdata(test_shot_number, filename=filename, data_fetcher='pyfusion.acquisition.DSV.fetch.DSVMultiChannelTimeseriesFetcher')
def testdata(self): """read in a test file and compare returned data object to expected result""" filename = os.path.join(os.path.dirname(__file__), "test_dsv.dat") data = genfromtxt(filename, unpack=True) test_acq = DSVAcquisition() test_shot_number = 12345 test_data = test_acq.getdata(test_shot_number, filename=filename, data_fetcher='pyfusion.acquisition.DSV.fetch.DSVMultiChannelTimeseriesFetcher') assert_array_almost_equal(data[0], test_data.timebase) for i, channel_i in enumerate(data[1:]): assert_array_almost_equal(channel_i, test_data.signal[i]) self.assertEqual(test_shot_number, test_data.meta['shot'])