def setUp(self):
# The tests require a data file. We will download some data and put it
# in a test directory. It needs to be writable by the user who initiates
# the tests.
self.original_datastore_path = settings.DATASTORE_PATH
path = os.path.join(settings.TEST_DATASTORE_PATH, "analysissessions")
test_datastore.setup_test_datastore_directory(path)
# Download real data of stations 3201, 3202 and 3203 on 25 March 2011.
# There are 52 coincidences.
date = datetime.date(2011, 3, 25)
file = test_datastore.get_datafile_path(date)
if not os.path.exists(file):
test_datastore.download_data_station(3201, date, get_blobs=True)
test_datastore.download_data_station(3202, date, get_blobs=True)
test_datastore.download_data_station(3203, date, get_blobs=True)
self.assertTrue(os.path.exists(file))
try:
data = tables.open_file(file, "r")
except Exception:
self.assertTrue(False)
self.assertEqual(
len(data.root.hisparc.cluster_leiden.station_3201.events), 50547)
self.assertEqual(
len(data.root.hisparc.cluster_leiden.station_3202.events), 43176)
self.assertEqual(
len(data.root.hisparc.cluster_leiden.station_3203.events), 18268)
data.close()
super(LiveSessionTestCase, self).setUp()
# Start Selenium
self.driver = webdriver.Firefox()
def setUp(self):
# The tests require a data file. We will download some data and put it
# in a test directory. It needs to be writable by the user who initiates
# the tests.
self.original_datastore_path = settings.DATASTORE_PATH
path = os.path.join(settings.TEST_DATASTORE_PATH, "analysissessions")
test_datastore.setup_test_datastore_directory(path)
# Download real data of stations 3201, 3202 and 3203 on 25 March 2011.
# There are 52 coincidences.
date = datetime.date(2011, 3, 25)
file = test_datastore.get_datafile_path(date)
if not os.path.exists(file):
test_datastore.download_data_station(3201, date, get_blobs=True)
test_datastore.download_data_station(3202, date, get_blobs=True)
test_datastore.download_data_station(3203, date, get_blobs=True)
self.assertTrue(os.path.exists(file))
try:
data = tables.open_file(file, "r")
except Exception:
self.assertTrue(False)
self.assertEqual(len(data.root.hisparc.cluster_leiden.station_3201.events), 50547)
self.assertEqual(len(data.root.hisparc.cluster_leiden.station_3202.events), 43176)
self.assertEqual(len(data.root.hisparc.cluster_leiden.station_3203.events), 18268)
data.close()
super(LiveSessionTestCase, self).setUp()
# Start Selenium
self.driver = webdriver.Firefox()
def setUp(self):
# make progressbar(list) do nothing (i.e., return list)
self.progressbar_patcher = patch('progressbar.ProgressBar')
self.progressbar_mock = self.progressbar_patcher.start()
self.progressbar_mock.return_value.side_effect = lambda x: x
# Setup test datastore
# The tests require a data file. We will download some data and
# put it in a test directory. It needs to be writable by the
# user who initiates the tests.
self.original_datastore_path = settings.DATASTORE_PATH
self.original_esd_path = settings.ESD_PATH
path = os.path.join(settings.TEST_DATASTORE_PATH, "histograms")
test_datastore.setup_test_datastore_directory(path)
# Download data
# Download real data of station 501 and 502 on 7 July 2011
# Here is data where we can fit the pulseheight MPV,
# and we get two stations to find coincidences.
date = DATE1
file = test_datastore.get_datafile_path(date)
if not os.path.exists(file):
test_datastore.download_data_station(STATION1, date, get_blobs=True)
test_datastore.download_data_station(STATION2, date, get_blobs=True)
self.assertTrue(os.path.exists(file))
try:
data = tables.open_file(file, "r")
except Exception:
self.fail()
self.assertEqual(data.root.hisparc.cluster_amsterdam.station_501.events.nrows, 63322)
self.assertEqual(data.root.hisparc.cluster_amsterdam.station_501.weather.nrows, 26317)
self.assertEqual(data.root.hisparc.cluster_amsterdam.station_502.events.nrows, 55967)
data.close()
# Download real data of station 501 on 16 May 2012, which also
# include weather data and configuration update.
date = DATE2
file = test_datastore.get_datafile_path(date)
if not os.path.exists(file):
test_datastore.download_data_station(STATION1, date, get_blobs=True)
self.assertTrue(os.path.exists(file))
try:
data = tables.open_file(file, "r")
except Exception:
self.fail()
self.assertEqual(data.root.hisparc.cluster_amsterdam.station_501.events.nrows, 6843)
self.assertEqual(data.root.hisparc.cluster_amsterdam.station_501.weather.nrows, 25918)
self.assertEqual(data.root.hisparc.cluster_amsterdam.station_501.config.nrows, 1)
data.close()
# Download test data of station 99 on 4 November 2013.
# This contains 2 events and a configuration from this test station.
date = DATETEST
file = test_datastore.get_datafile_path(date)
if not os.path.exists(file):
test_datastore.download_data_station(STATIONTEST, date, get_blobs=True)
self.assertTrue(os.path.exists(file))
try:
data = tables.open_file(file, "r")
except Exception:
self.fail()
self.assertEqual(data.root.hisparc.cluster_amsterdam.station_99.events.nrows, 2)
self.assertEqual(data.root.hisparc.cluster_amsterdam.station_99.config.nrows, 1)
data.close()
# Reset last updated state
# Reset generator state such that it will always update when new
# files are found.
state = GeneratorState.objects.get()
state.update_last_run = datetime.datetime.fromtimestamp(0)
state.check_last_run = datetime.datetime.fromtimestamp(0)
state.save()
super(BaseHistogramsTestCase, self).setUp()
