def test_get_area(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
a = np.array(
np.meshgrid(np.linspace(5, 10, 10),
np.linspace(-150, -160, 10)))
r = n.get_area(a, 0, 0, 'votemper')
self.assertAlmostEqual(r[5, 5], 301.28, places=2)
def test_get_profile(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
p, d = n.get_profile(13.0, -149.0, 0, 'votemper')
self.assertAlmostEqual(p[0], 299.18, places=2)
self.assertAlmostEqual(p[10], 299.16, places=2)
self.assertAlmostEqual(p[20], 296.46, places=2)
self.assertTrue(np.ma.is_masked(p[49]))
def open_dataset(url):
if url is not None:
if __dataset_cache.get(url) is None:
if url.startswith("http") or url.endswith(".nc"):
with Dataset(url, 'r') as ds:
if 'latitude_longitude' in ds.variables or \
'LatLon_Projection' in ds.variables:
__dataset_cache[url] = mercator.Mercator(url)
elif 'siglay' in ds.variables:
__dataset_cache[url] = fvcom.Fvcom(url)
elif 'polar_stereographic' in ds.variables:
__dataset_cache[url] = nemo.Nemo(url)
else:
__dataset_cache[url] = nemo.Nemo(url)
return __dataset_cache.get(url)
def test_get_raw_point(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
lat, lon, data = n.get_raw_point(13.0, -149.0, 0, 0, 'votemper')
self.assertEqual(len(lat.ravel()), 12)
self.assertEqual(len(lon.ravel()), 12)
self.assertEqual(len(data.ravel()), 12)
self.assertAlmostEqual(data[1, 1], 299.3, places=1)
def test_variables(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
variables = n.variables
self.assertEqual(len(variables), 5)
self.assertTrue('votemper' in variables)
self.assertEqual(variables['votemper'].name,
'Water temperature at CMC')
self.assertEqual(variables['votemper'].unit, 'Kelvins')
def test_get_timeseries_profile(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
r, d = n.get_timeseries_profile(13.0, -149.0, 0, 1, 'votemper')
self.assertAlmostEqual(r[0, 0], 299.18, places=2)
self.assertAlmostEqual(r[0, 10], 299.16, places=2)
self.assertAlmostEqual(r[0, 20], 296.46, places=2)
self.assertTrue(np.ma.is_masked(r[0, 49]))
self.assertNotEqual(r[0, 0], r[1, 0])
self.assertTrue(np.ma.is_masked(r[1, 49]))
def test_get_path_profile(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
p, d, r, dep = n.get_path_profile(
[[13, -149], [14, -140], [15, -130]], 0, 'votemper', 10)
self.assertEqual(r.shape[0], 50)
self.assertGreater(r.shape[1], 10)
self.assertEqual(r.shape[1], p.shape[1])
self.assertEqual(r.shape[1], len(d))
self.assertEqual(d[0], 0)
def test_get_profile_depths(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
p = n.get_profile_depths(
13.0,
-149.0,
0,
'votemper',
[0, 10, 25, 50, 100, 200, 500, 1000]
)
self.assertTrue(np.ma.is_masked(p[0]))
self.assertAlmostEqual(p[1], 299.17, places=2)
self.assertAlmostEqual(p[4], 292.47, places=2)
self.assertAlmostEqual(p[7], 277.90, places=2)
def test_timestamps(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
self.assertEqual(len(n.timestamps), 2)
self.assertEqual(
n.timestamps[0],
datetime.datetime(2014, 5, 17, 0, 0, 0, 0, pytz.UTC))
# Property is read-only
with self.assertRaises(AttributeError):
n.timestamps = []
# List is immutable
with self.assertRaises(ValueError):
n.timestamps[0] = 0
def test_init(self):
nemo.Nemo(None)
def test_get_timeseries_point(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
r = n.get_timeseries_point(13.0, -149.0, 0, 0, 1, 'votemper')
self.assertAlmostEqual(r[0], 299.18, places=2)
self.assertAlmostEqual(r[1], 299.72, places=2)
def test_bottom_point(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
self.assertAlmostEqual(n.get_point(13.0, -149.0, 'bottom', 0,
'votemper'),
274.13,
places=2)
def test_get_point(self):
with nemo.Nemo('data/testdata/nemo_test.nc') as n:
self.assertAlmostEqual(n.get_point(13.0, -149.0, 0, 0, 'votemper'),
299.18,
places=2)
def test_open(self):
with nemo.Nemo('data/testdata/nemo_test.nc'):
pass
#!/usr/bin/env python3
import asyncio
import nemo
import helper
import config
import discord
from discord.ext.commands import has_permissions
number_emojis = ["1⃣", "2⃣", "3⃣", "4⃣", "5⃣", "6⃣", "7⃣", "8⃣", "9⃣"]
nemo = nemo.Nemo()
@nemo.command("!")
@has_permissions(administrator=True)
async def setup(*, message: discord.Message, guild: discord.Guild, **_):
if any(x.name == config.CATEGORY_NAME and x.type == discord.ChannelType.category for x in guild.channels):
return
category = await guild.create_category(config.CATEGORY_NAME)
organization = await guild.create_text_channel(config.ORGANIZATION_NAME, overwrites={
guild.default_role: discord.PermissionOverwrite(send_messages=False),
guild.me: discord.PermissionOverwrite(send_messages=True)
}, category=category)
await help_msg(organization)
await message.delete()
@nemo.command("!help")
@has_permissions(administrator=True)
@helper.auto_delete
