def test_set_shape(self):
# Define some arbitrary data
current = dimpy.data([1.0, 2.0, 3.0], units='A')
volume = dimpy.data([1.0], units='mL')
mass = dimpy.data([1.0], units='kg')
acceleration = dimpy.data([1.0], units='m/s^2')
force = dimpy.data([1.0], units='N')
# Check that the dimensional set shapes are correct
self.assertEquals(dimpy.analysis.get_dimensional_set_shape([mass, acceleration, force]), (3, 3, 1))
self.assertEquals(dimpy.analysis.get_dimensional_set_shape([mass, acceleration, force, current]), (4, 4, 1))
self.assertEquals(dimpy.analysis.get_dimensional_set_shape([mass, acceleration, force, volume]), (4, 3, 1))
self.assertEquals(dimpy.analysis.get_dimensional_set_shape([force, current]), (2, 4, 0))
self.assertEquals(dimpy.analysis.get_dimensional_set_shape([force, volume]), (2, 3, 0))
def test_set_shape(self):
# Define some arbitrary data
current = dimpy.data([1.0, 2.0, 3.0], units='A')
volume = dimpy.data([1.0], units='mL')
mass = dimpy.data([1.0], units='kg')
acceleration = dimpy.data([1.0], units='m/s^2')
force = dimpy.data([1.0], units='N')
# Check that the dimensional set shapes are correct
self.assertEquals(
dimpy.analysis.get_dimensional_set_shape(
[mass, acceleration, force]), (3, 3, 1))
self.assertEquals(
dimpy.analysis.get_dimensional_set_shape(
[mass, acceleration, force, current]), (4, 4, 1))
self.assertEquals(
dimpy.analysis.get_dimensional_set_shape(
[mass, acceleration, force, volume]), (4, 3, 1))
self.assertEquals(
dimpy.analysis.get_dimensional_set_shape([force, current]),
(2, 4, 0))
self.assertEquals(
dimpy.analysis.get_dimensional_set_shape([force, volume]),
(2, 3, 0))
def test_dimensional_homogeneity(self):
# Check that you cannot add dissimilar dimensions
with self.assertRaises(dimpy.DimensionError):
dimpy.data([1.0], units='m') + dimpy.data([1.0], units='s')
# Check that you can't take the sine of a dimensioned value
with self.assertRaises(dimpy.DimensionError):
np.sin(dimpy.data([1.0], units='m'))
# Check that you can't take the square root of an odd dimension
with self.assertRaises(dimpy.DimensionError):
np.sqrt(dimpy.data([1.0], units='L'))
# Check that you can't raise a value to a non-integer value
with self.assertRaises(dimpy.DimensionError):
dimpy.data([1.0], units='m') ** 1.1
def test_dimension_arithmetic(self):
# Check that multiplication results in summation of dimensions
mass = dimpy.data([1.0], units='kg')
acceleration = dimpy.data([1.0], units='m/s^2')
force = dimpy.data([1.0], units='N')
np.testing.assert_array_almost_equal(mass * acceleration, force)
np.testing.assert_array_almost_equal(force / acceleration, mass)
np.testing.assert_array_almost_equal(force / mass, acceleration)
# Check that taking the square root halves the dimensions, and that squaring doubles them
area = dimpy.data([100.0], units='m^2')
side = dimpy.data([10.0], units='m')
np.testing.assert_array_almost_equal(np.sqrt(area), side)
np.testing.assert_array_almost_equal(np.square(side), area)
# Check that raising a value to a power handles the dimensions properly
length = dimpy.data([0.1, 1.0, 10.0], units='m')
volume = dimpy.data([1.0, 1000.0, 1000000.0], units='L')
np.testing.assert_array_almost_equal(length**3, volume)
# Check other equality behavior
self.assertTrue(dimpy.data([1.0], units='mm') < dimpy.data([1.0], units='m'))
self.assertTrue(dimpy.data([1.0], units='mm') <= dimpy.data([1.0], units='m'))
self.assertTrue(dimpy.data([1.0], units='m') > dimpy.data([1.0], units='mm'))
self.assertTrue(dimpy.data([1.0], units='m') >= dimpy.data([1.0], units='mm'))
def test_scalars(self):
# Check that we can grab a single element from an array
lengths = dimpy.data([1.0, 2.0, 3.0], units='m')
self.assertEqual(lengths[0], dimpy.data([1.0], units='m'))
self.assertEqual(lengths[1], dimpy.data([2.0], units='m'))
self.assertEqual(lengths[2], dimpy.data([3.0], units='m'))
def test_powers(self):
# Check that arrays scale properly to the native internal representation
np.testing.assert_array_almost_equal(dimpy.data([1.0, 2.0, 3.0], units='mm'), dimpy.data([1e-3, 2e-3, 3e-3], units='m'))
def test_repr(self):
# Check that the array repr matches intended output
self.assertEquals(repr(dimpy.data([1.0, 2.0, 3.0], units='N')),
'[ 1. 2. 3.] - dimensions: [mass: 1, length: 1, time: -2, current: 0, temperature: 0, amount of substance: 0, luminous intensity: 0]')
# -*- coding: utf-8 -*-
"""
Created on Sun Jun 05 17:56:27 2016
@author: Matthew Fulghum
"""
import numpy as np
import dimpy
vel = dimpy.data([1.0, 2.0, 3.0], units='mm/s')
area = dimpy.data([20.0], units='m^2')
mass = dimpy.data([1.0], units='kg')
acceleration = dimpy.data([1.0], units='m/s^2')
force = dimpy.data([1.0], units='N')
length = dimpy.data([1.0], units='cm')
volume = dimpy.data([1.0], units='mL')
# Pendulum model
T = dimpy.unit('s')
M = dimpy.unit('kg')
L = dimpy.unit('m')
g = dimpy.unit('m/s^2')
variables = [T, M, L, g]
print(dimpy.analyze(variables))
# Black body model
u = dimpy.unit('kg/(m^2-s^2)')