feat: Add complex numbers
No negation (and therefore no subtraction) since that needs self-reference.
This commit is contained in:
parent
b59a8c2ca9
commit
b82336e590
14
complex/Complex.mjs
Normal file
14
complex/Complex.mjs
Normal file
@ -0,0 +1,14 @@
|
||||
import typed from 'typed-function'
|
||||
|
||||
/* Use a plain object with keys re and im for a complex */
|
||||
typed.addType({
|
||||
name: 'Complex',
|
||||
test: z => z && typeof z === 'object' && 're' in z && 'im' in z
|
||||
})
|
||||
|
||||
typed.addConversion({
|
||||
from: 'number',
|
||||
to: 'Complex',
|
||||
convert: x => ({re: x, im: 0})
|
||||
})
|
||||
|
12
complex/add.mjs
Normal file
12
complex/add.mjs
Normal file
@ -0,0 +1,12 @@
|
||||
import './Complex.mjs'
|
||||
|
||||
export const add = {
|
||||
'...Complex': [[], addends => {
|
||||
const sum = {re: 0, im: 0}
|
||||
addends.forEach(addend => {
|
||||
sum.re += addend.re
|
||||
sum.im += addend.im
|
||||
})
|
||||
return sum
|
||||
}]
|
||||
}
|
2
complex/all.mjs
Normal file
2
complex/all.mjs
Normal file
@ -0,0 +1,2 @@
|
||||
export {complex} from './complex.mjs'
|
||||
export {add} from './add.mjs'
|
6
complex/complex.mjs
Normal file
6
complex/complex.mjs
Normal file
@ -0,0 +1,6 @@
|
||||
import './Complex.mjs'
|
||||
|
||||
export const complex = {
|
||||
'number, number': [[], (x, y) => ({re: x, im: y})],
|
||||
Complex: [[], z => z]
|
||||
}
|
@ -1,8 +1,10 @@
|
||||
/* Core of pocomath: generates the default instance */
|
||||
import PocomathInstance from './PocomathInstance.mjs'
|
||||
import * as numbers from './number/all.mjs'
|
||||
import * as complex from './complex/all.mjs'
|
||||
|
||||
const math = new PocomathInstance('math')
|
||||
math.install(numbers)
|
||||
math.install(complex)
|
||||
|
||||
export default math
|
||||
|
@ -16,4 +16,18 @@ describe('The default full pocomath instance "math"', () => {
|
||||
assert.strictEqual(math.negate(-1), 1)
|
||||
assert.strictEqual(math.add(10, math.negate(3)), 7)
|
||||
})
|
||||
|
||||
it('can be extended', () => {
|
||||
math.install({'add': {
|
||||
'...string': [[], addends => addends.reduce((x,y) => x+y, '')]
|
||||
}})
|
||||
assert.strictEqual(math.add('Kilroy',' is here'), 'Kilroy is here')
|
||||
})
|
||||
|
||||
it('handles complex numbers', () => {
|
||||
const norm13 = {re: 2, im: 3}
|
||||
assert.deepStrictEqual(math.complex(2,3), norm13)
|
||||
assert.deepStrictEqual(math.complex(2), math.complex(2,0))
|
||||
assert.deepStrictEqual(math.add(2, math.complex(0,3)), norm13)
|
||||
})
|
||||
})
|
||||
|
21
test/custom.mjs
Normal file
21
test/custom.mjs
Normal file
@ -0,0 +1,21 @@
|
||||
import assert from 'assert'
|
||||
import typed from 'typed-function'
|
||||
import PocomathInstance from '../PocomathInstance.mjs'
|
||||
import * as numbers from '../number/all.mjs'
|
||||
import * as complex from '../complex/all.mjs'
|
||||
|
||||
const bw = new PocomathInstance('backwards')
|
||||
describe('A custom instance', () => {
|
||||
it("works when partially assembled", () => {
|
||||
bw.install(complex)
|
||||
assert.deepStrictEqual(bw.add(2, bw.complex(0, 3)), {re: 2, im: 3})
|
||||
})
|
||||
|
||||
it("can be assembled in any order", () => {
|
||||
bw.install(numbers)
|
||||
typed.addConversion({from: 'string', to: 'number', convert: x => +x})
|
||||
assert.strictEqual(bw.subtract(16, bw.add(3,4,2)), 7)
|
||||
assert.strictEqual(bw.negate('8'), -8)
|
||||
assert.deepStrictEqual(bw.add(bw.complex(1,3), 1), {re: 2, im: 3})
|
||||
})
|
||||
})
|
Loading…
Reference in New Issue
Block a user